diff --git a/README.md b/README.md
index 07700f3..6cd1e30 100644
--- a/README.md
+++ b/README.md
@@ -37,7 +37,7 @@ like [Numpy](https://numpy.org) and [Pillow](https://pypi.org/project/pillow/) f
    ```
 ### Generating an image
 
-Run the provided [main.py](main.py) by specifying a prompt and some optional arguments like so using the `schnell` model:
+Run the provided [main.py](src/mflux/main.py) by specifying a prompt and some optional arguments like so using the `schnell` model:
 
 ```
 python main.py --model schnell --prompt "Luxury food photograph" --steps 2 --seed 2 -q 8
@@ -116,7 +116,7 @@ image = flux.generate_image(
 image.save(path="image.png")
 ```
 
-For more options on how to configure MFLUX, please see [main.py](main.py).
+For more options on how to configure MFLUX, please see [main.py](src/mflux/main.py).
 
 ### Image generation speed (updated)
 
diff --git a/main.py b/src/mflux/main.py
similarity index 100%
rename from main.py
rename to src/mflux/main.py
diff --git a/save.py b/src/mflux/save.py
similarity index 100%
rename from save.py
rename to src/mflux/save.py