diff --git a/labs/lab_1/lab_1a.py b/labs/lab_1/lab_1a.py
index 9d15ec83..94f6425e 100644
--- a/labs/lab_1/lab_1a.py
+++ b/labs/lab_1/lab_1a.py
@@ -3,14 +3,17 @@
 
 The first lab in the BWSI CSS course. To complete this lab, fill out the variable on line 10
 with your name. Then, save the code, add it to the staging area, and commit it to the Git tree.
+
+This is to simulate a change made on a robot: robot_speed = 5 # m/s
 """
 
 def main():
     print("Hello World!")
 
-    name = "" # TODO: Insert your name between the double quotes
+    name = "shanil ashraf" # TODO: Insert your name between the double quotes
 
     print(f"{name}, Welcome to the CSS course!")
+    print("I like playing badminton w my friends")
 
 if __name__ == "__main__":
     main()
diff --git a/labs/lab_1/lab_1b.py b/labs/lab_1/lab_1b.py
index e58dd957..01f463c9 100644
--- a/labs/lab_1/lab_1b.py
+++ b/labs/lab_1/lab_1b.py
@@ -23,19 +23,12 @@ def simple_calculator(operation: str, num1: float, num2: float) -> float:
         float: The result of the operation.
     """
 
-    if operation == "add":
-        return num1 + num2
-    elif operation == "subtract":
-        return num1 - num2
-    elif operation == "multiply":
-        return num1 * num2
-    elif operation == "divide":
-        if num2 != 0:
-            return num1 / num2
-        else:
-            raise ValueError("Cannot divide by zero.")
-    else:
-        raise ValueError("Invalid operation. Please choose from 'add', 'subtract', 'multiply', or 'divide'.")
+    while True:
+        try:
+            number = float(input(prompt))
+            return number
+        except ValueError:
+            print("Invalid input. Please enter a valid number.")
 
 def main():