diff --git a/reinforce_cartpole.py b/reinforce_cartpole.py
index 541eeef899b637842836a5d80a1e1ac144e753e7..a76752443ce105985a35b08d86a9b38c2f639d97 100644
--- a/reinforce_cartpole.py
+++ b/reinforce_cartpole.py
@@ -1,3 +1,20 @@
+import gymnasium as gym
+import torch
+import torch.nn as nn
+import torch.optim as optim
+import torch.nn.functional as F
+from torch.distributions import Categorical
+import matplotlib.pyplot as plt
+
+
+
+
+
+
+
+
+
+
 # Create the environment
 env = gym.make("CartPole-v1", render_mode="human")