diff --git a/utils/schedule_solver/solver.py b/utils/schedule_solver/solver.py
index 98a87974..f7d5c03b 100644
--- a/utils/schedule_solver/solver.py
+++ b/utils/schedule_solver/solver.py
@@ -38,13 +38,16 @@ def main():
     req_hours = [[0 for x in range(len(hours))] for y in range(len(days))]
 
     # get arguments
-    # there will be 7 arguments, monday to sunday schedule
+    # there will be 8 arguments, monday to sunday schedule and the shift selected
+    # sample argument: 
+    # 0-0-0-0-0-0-0-0-0-0-0-0-0-0-0-0-0-0-0-0-0-0-0-0 0-0-0-0-0-0-0-0-0-0-0-0-0-0-0-0-0-0-0-0-0-0-0-0 0-0-0-0-0-0-0-0-0-0-0-0-0-0-0-0-0-0-0-0-0-0-0-0 0-0-0-0-0-0-0-0-0-0-0-0-0-0-0-0-0-0-0-0-0-0-0-0 0-0-0-0-0-0-0-0-0-0-0-0-0-0-0-0-0-0-0-0-0-0-0-0 0-0-0-0-0-0-0-0-0-0-0-0-0-0-0-0-0-0-0-0-0-0-0-0 0-0-0-0-0-0-0-0-0-0-0-0-0-0-0-0-0-0-0-0-0-0-0-0 24_7
     for day_index in range(0, len(days)):
         hours_string = sys.argv[day_index + 1]
         hours_data = hours_string.split('-')
         for hour_index in range(0, len(hours)):
             req_hours[day_index][hour_index] = int(hours_data[hour_index])
 
+    # index of shift selected is 8
     shift = sys.argv[8]
     hour_shifts = []
     if shift == "24_7":