diff --git a/main.py b/main.py
index 8b26c5b..51b12ea 100644
--- a/main.py
+++ b/main.py
@@ -8,6 +8,8 @@ from flux_1_schnell.flux import Flux1
 from flux_1_schnell.post_processing.image_util import ImageUtil
 
 flux = Flux1("black-forest-labs/FLUX.1-schnell", max_sequence_length=256)
+# flux = Flux1("black-forest-labs/FLUX.1-dev", max_sequence_length=512)
+
 
 image = flux.generate_image(
     seed=3,
@@ -16,6 +18,7 @@ image = flux.generate_image(
         num_inference_steps=2,
         height=768,
         width=1360,
+        guidance=3.5,
     )
 )
 
diff --git a/main_dev.py b/main_dev.py
deleted file mode 100644
index 6d24b9b..0000000
--- a/main_dev.py
+++ /dev/null
@@ -1,22 +0,0 @@
-import os
-import sys
-
-sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), 'src')))
-
-from flux_1_schnell.config.config import Config
-from flux_1_schnell.flux import Flux1
-
-flux = Flux1("black-forest-labs/FLUX.1-dev", max_sequence_length=512)
-
-image = flux.generate_image(
-    seed=3,
-    prompt="Luxury food photograph of a birthday cake. In the middle it has three candles shaped like letters spelling the word 'MLX'. It has perfect lighting and a cozy background with big bokeh and shallow depth of field. The mood is a sunset balcony in tuscany. The photo is taken from the side of the cake. The scene is complemented by a warm, inviting light that highlights the textures and colors of the ingredients, giving it an appetizing and elegant look.",
-    config=Config(
-        num_inference_steps=20,
-        height=768,
-        width=1360,
-        guidance=3.5,
-    )
-)
-
-image.save("image.png")