diff --git a/main.py b/main.py
index c5fc989777f8a054c17609910c9faa8206da5375..a4257f82ff5420373e854f563eb86b6cc78804ef 100644
--- a/main.py
+++ b/main.py
@@ -39,8 +39,10 @@ while is_log_heater != 0 and is_log_heater != 1:
         print()
 
     try:
+        # Attempt to convert the input to int and store it.
         is_log_heater = int(input("measurement process = "))
     except ValueError:
+        # If conversion fails, set value to 2, in order to ask user again.
         is_log_heater = 2
 
 # Read the metadata necessary to control test rig from the setup file.