diff --git a/s487175-create-dataset-script.py b/s487175-create-dataset-script.py
index e107e48..d9642ec 100644
--- a/s487175-create-dataset-script.py
+++ b/s487175-create-dataset-script.py
@@ -1,13 +1,18 @@
 
 #!/usr/bin/env python
+
 import os
 import pandas as pd
 diamonds = pd.read_csv('diamonds.csv')
-#Wyświetlenie zbioru danych
-diamonds
 
-cutoff = int(os.environ['CUTOFF'])
-diamonds = diamonds[:cutoff]
+#Wyświetlenie zbioru danych
+cutoff = int(os.environ.get('CUTOFF', -1))
+if cutoff == -1:
+    print("CUTOFF parameter not set.")
+    diamonds
+else:
+    cutoff = int(os.environ['CUTOFF'])
+    diamonds = diamonds[:cutoff]
 
 # %%
 #przydzielanie nazwy kolumny z id