diff --git a/s487175-create-dataset-script.py b/s487175-create-dataset-script.py
index 055700a..b55c834 100644
--- a/s487175-create-dataset-script.py
+++ b/s487175-create-dataset-script.py
@@ -9,11 +9,16 @@ diamonds = pd.read_csv('diamonds.csv')
 cutoff = os.environ.get('CUTOFF', None)
 if cutoff is None:
     print("CUTOFF parameter not set.")
-else:
-    cutoff = int(cutoff)
-    diamonds = diamonds[:cutoff]
-    diamonds.to_csv('diamonds.csv', index=False) #nadpisanie pliku
     diamonds
+else:
+    if cutoff.strip() == "":
+        print("CUTOFF parameter is empty.")
+        diamonds
+    else:
+        cutoff = int(cutoff)
+        diamonds = diamonds[:cutoff]
+        diamonds.to_csv('diamonds.csv', index=False) ## nadpisanie pliku
+        diamonds
 
 # %%
 #przydzielanie nazwy kolumny z id