diff --git a/s487175-create-dataset-script.py b/s487175-create-dataset-script.py
index 10229a9..e107e48 100644
--- a/s487175-create-dataset-script.py
+++ b/s487175-create-dataset-script.py
@@ -1,11 +1,14 @@
 
 #!/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]
+
 # %%
 #przydzielanie nazwy kolumny z id
 diamonds = diamonds.rename(columns={diamonds.columns[0]: 'id'})