diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
index 5d16d87..ffc13b4 100644
--- a/.github/workflows/ci.yml
+++ b/.github/workflows/ci.yml
@@ -27,12 +27,12 @@ jobs:
 
             - name: Lint with Ruff
               run: |
-                  ruff check src/fastfiz_env
-                  ruff format src/fastfiz_env
+                  ruff check fastfiz_env
+                  ruff format fastfiz_env
 
             - name: Run MyPy
               run: |
-                  mypy src/fastfiz_env
+                  mypy fastfiz_env
 
             - name: Test with pytest
               run: |
diff --git a/src/fastfiz_env/__init__.py b/fastfiz_env/__init__.py
similarity index 100%
rename from src/fastfiz_env/__init__.py
rename to fastfiz_env/__init__.py
diff --git a/src/fastfiz_env/envs/__init__.py b/fastfiz_env/envs/__init__.py
similarity index 100%
rename from src/fastfiz_env/envs/__init__.py
rename to fastfiz_env/envs/__init__.py
diff --git a/src/fastfiz_env/envs/fastfiz.py b/fastfiz_env/envs/fastfiz.py
similarity index 100%
rename from src/fastfiz_env/envs/fastfiz.py
rename to fastfiz_env/envs/fastfiz.py
diff --git a/src/fastfiz_env/envs/pockets_fastfiz.py b/fastfiz_env/envs/pockets_fastfiz.py
similarity index 100%
rename from src/fastfiz_env/envs/pockets_fastfiz.py
rename to fastfiz_env/envs/pockets_fastfiz.py
diff --git a/src/fastfiz_env/envs/utils.py b/fastfiz_env/envs/utils.py
similarity index 100%
rename from src/fastfiz_env/envs/utils.py
rename to fastfiz_env/envs/utils.py
diff --git a/src/fastfiz_env/make.py b/fastfiz_env/make.py
similarity index 100%
rename from src/fastfiz_env/make.py
rename to fastfiz_env/make.py
diff --git a/src/fastfiz_env/reward_functions/__init__.py b/fastfiz_env/reward_functions/__init__.py
similarity index 100%
rename from src/fastfiz_env/reward_functions/__init__.py
rename to fastfiz_env/reward_functions/__init__.py
diff --git a/src/fastfiz_env/reward_functions/binary_reward.py b/fastfiz_env/reward_functions/binary_reward.py
similarity index 100%
rename from src/fastfiz_env/reward_functions/binary_reward.py
rename to fastfiz_env/reward_functions/binary_reward.py
diff --git a/src/fastfiz_env/reward_functions/combined_reward.py b/fastfiz_env/reward_functions/combined_reward.py
similarity index 100%
rename from src/fastfiz_env/reward_functions/combined_reward.py
rename to fastfiz_env/reward_functions/combined_reward.py
diff --git a/src/fastfiz_env/reward_functions/common/__init__.py b/fastfiz_env/reward_functions/common/__init__.py
similarity index 100%
rename from src/fastfiz_env/reward_functions/common/__init__.py
rename to fastfiz_env/reward_functions/common/__init__.py
diff --git a/src/fastfiz_env/reward_functions/common/balls_not_moved_reward.py b/fastfiz_env/reward_functions/common/balls_not_moved_reward.py
similarity index 100%
rename from src/fastfiz_env/reward_functions/common/balls_not_moved_reward.py
rename to fastfiz_env/reward_functions/common/balls_not_moved_reward.py
diff --git a/src/fastfiz_env/reward_functions/common/constant_reward.py b/fastfiz_env/reward_functions/common/constant_reward.py
similarity index 100%
rename from src/fastfiz_env/reward_functions/common/constant_reward.py
rename to fastfiz_env/reward_functions/common/constant_reward.py
diff --git a/src/fastfiz_env/reward_functions/common/cue_ball_not_moved_reward.py b/fastfiz_env/reward_functions/common/cue_ball_not_moved_reward.py
similarity index 100%
rename from src/fastfiz_env/reward_functions/common/cue_ball_not_moved_reward.py
rename to fastfiz_env/reward_functions/common/cue_ball_not_moved_reward.py
diff --git a/src/fastfiz_env/reward_functions/common/cue_ball_pocketed_reward.py b/fastfiz_env/reward_functions/common/cue_ball_pocketed_reward.py
similarity index 100%
rename from src/fastfiz_env/reward_functions/common/cue_ball_pocketed_reward.py
rename to fastfiz_env/reward_functions/common/cue_ball_pocketed_reward.py
diff --git a/src/fastfiz_env/reward_functions/common/delta_best_total_distance_reward.py b/fastfiz_env/reward_functions/common/delta_best_total_distance_reward.py
similarity index 100%
rename from src/fastfiz_env/reward_functions/common/delta_best_total_distance_reward.py
rename to fastfiz_env/reward_functions/common/delta_best_total_distance_reward.py
diff --git a/src/fastfiz_env/reward_functions/common/exponential_velocity_reward.py b/fastfiz_env/reward_functions/common/exponential_velocity_reward.py
similarity index 100%
rename from src/fastfiz_env/reward_functions/common/exponential_velocity_reward.py
rename to fastfiz_env/reward_functions/common/exponential_velocity_reward.py
diff --git a/src/fastfiz_env/reward_functions/common/game_won_reward.py b/fastfiz_env/reward_functions/common/game_won_reward.py
similarity index 100%
rename from src/fastfiz_env/reward_functions/common/game_won_reward.py
rename to fastfiz_env/reward_functions/common/game_won_reward.py
diff --git a/src/fastfiz_env/reward_functions/common/impossible_shot_reward.py b/fastfiz_env/reward_functions/common/impossible_shot_reward.py
similarity index 100%
rename from src/fastfiz_env/reward_functions/common/impossible_shot_reward.py
rename to fastfiz_env/reward_functions/common/impossible_shot_reward.py
diff --git a/src/fastfiz_env/reward_functions/common/step_no_balls_pocketed_reward.py b/fastfiz_env/reward_functions/common/step_no_balls_pocketed_reward.py
similarity index 100%
rename from src/fastfiz_env/reward_functions/common/step_no_balls_pocketed_reward.py
rename to fastfiz_env/reward_functions/common/step_no_balls_pocketed_reward.py
diff --git a/src/fastfiz_env/reward_functions/common/step_pocketed_reward.py b/fastfiz_env/reward_functions/common/step_pocketed_reward.py
similarity index 100%
rename from src/fastfiz_env/reward_functions/common/step_pocketed_reward.py
rename to fastfiz_env/reward_functions/common/step_pocketed_reward.py
diff --git a/src/fastfiz_env/reward_functions/common/total_distance_reward.py b/fastfiz_env/reward_functions/common/total_distance_reward.py
similarity index 100%
rename from src/fastfiz_env/reward_functions/common/total_distance_reward.py
rename to fastfiz_env/reward_functions/common/total_distance_reward.py
diff --git a/src/fastfiz_env/reward_functions/common/velocity_reward.py b/fastfiz_env/reward_functions/common/velocity_reward.py
similarity index 100%
rename from src/fastfiz_env/reward_functions/common/velocity_reward.py
rename to fastfiz_env/reward_functions/common/velocity_reward.py
diff --git a/src/fastfiz_env/reward_functions/common/weights.py b/fastfiz_env/reward_functions/common/weights.py
similarity index 100%
rename from src/fastfiz_env/reward_functions/common/weights.py
rename to fastfiz_env/reward_functions/common/weights.py
diff --git a/src/fastfiz_env/reward_functions/default_reward.py b/fastfiz_env/reward_functions/default_reward.py
similarity index 100%
rename from src/fastfiz_env/reward_functions/default_reward.py
rename to fastfiz_env/reward_functions/default_reward.py
diff --git a/src/fastfiz_env/reward_functions/reward_function.py b/fastfiz_env/reward_functions/reward_function.py
similarity index 100%
rename from src/fastfiz_env/reward_functions/reward_function.py
rename to fastfiz_env/reward_functions/reward_function.py
diff --git a/src/fastfiz_env/reward_functions/winning_reward.py b/fastfiz_env/reward_functions/winning_reward.py
similarity index 100%
rename from src/fastfiz_env/reward_functions/winning_reward.py
rename to fastfiz_env/reward_functions/winning_reward.py
diff --git a/src/fastfiz_env/utils/__init__.py b/fastfiz_env/utils/__init__.py
similarity index 100%
rename from src/fastfiz_env/utils/__init__.py
rename to fastfiz_env/utils/__init__.py
diff --git a/src/fastfiz_env/utils/fastfiz/__init__.py b/fastfiz_env/utils/fastfiz/__init__.py
similarity index 100%
rename from src/fastfiz_env/utils/fastfiz/__init__.py
rename to fastfiz_env/utils/fastfiz/__init__.py
diff --git a/src/fastfiz_env/utils/fastfiz/fastfiz.py b/fastfiz_env/utils/fastfiz/fastfiz.py
similarity index 100%
rename from src/fastfiz_env/utils/fastfiz/fastfiz.py
rename to fastfiz_env/utils/fastfiz/fastfiz.py
diff --git a/src/fastfiz_env/wrappers/__init__.py b/fastfiz_env/wrappers/__init__.py
similarity index 100%
rename from src/fastfiz_env/wrappers/__init__.py
rename to fastfiz_env/wrappers/__init__.py
diff --git a/src/fastfiz_env/wrappers/action.py b/fastfiz_env/wrappers/action.py
similarity index 100%
rename from src/fastfiz_env/wrappers/action.py
rename to fastfiz_env/wrappers/action.py
diff --git a/src/fastfiz_env/wrappers/time_limit_injection.py b/fastfiz_env/wrappers/time_limit_injection.py
similarity index 100%
rename from src/fastfiz_env/wrappers/time_limit_injection.py
rename to fastfiz_env/wrappers/time_limit_injection.py
diff --git a/src/fastfiz_env/wrappers/utils.py b/fastfiz_env/wrappers/utils.py
similarity index 100%
rename from src/fastfiz_env/wrappers/utils.py
rename to fastfiz_env/wrappers/utils.py
diff --git a/src/eval.py b/scripts/eval.py
similarity index 100%
rename from src/eval.py
rename to scripts/eval.py
diff --git a/src/hyperparams.py b/scripts/hyperparams.py
similarity index 100%
rename from src/hyperparams.py
rename to scripts/hyperparams.py
diff --git a/src/optimize.py b/scripts/optimize.py
similarity index 100%
rename from src/optimize.py
rename to scripts/optimize.py
diff --git a/src/plot.py b/scripts/plot.py
similarity index 100%
rename from src/plot.py
rename to scripts/plot.py
diff --git a/src/random_policy.py b/scripts/random_policy.py
similarity index 100%
rename from src/random_policy.py
rename to scripts/random_policy.py
diff --git a/src/train.py b/scripts/train.py
similarity index 100%
rename from src/train.py
rename to scripts/train.py
diff --git a/src/tests/__init__.py b/tests/__init__.py
similarity index 100%
rename from src/tests/__init__.py
rename to tests/__init__.py
diff --git a/src/tests/envs/__init__.py b/tests/envs/__init__.py
similarity index 100%
rename from src/tests/envs/__init__.py
rename to tests/envs/__init__.py
diff --git a/src/tests/envs/test_envs.py b/tests/envs/test_envs.py
similarity index 100%
rename from src/tests/envs/test_envs.py
rename to tests/envs/test_envs.py
diff --git a/src/tests/utils/__init__.py b/tests/utils/__init__.py
similarity index 100%
rename from src/tests/utils/__init__.py
rename to tests/utils/__init__.py
diff --git a/src/tests/utils/test_fastfiz.py b/tests/utils/test_fastfiz.py
similarity index 100%
rename from src/tests/utils/test_fastfiz.py
rename to tests/utils/test_fastfiz.py
diff --git a/src/tests/utils/test_reward_functions.py b/tests/utils/test_reward_functions.py
similarity index 100%
rename from src/tests/utils/test_reward_functions.py
rename to tests/utils/test_reward_functions.py
diff --git a/src/tests/utils/test_wrappers.py b/tests/utils/test_wrappers.py
similarity index 100%
rename from src/tests/utils/test_wrappers.py
rename to tests/utils/test_wrappers.py