Update caught_speeding.py

Author: michael-bateman

logic-1/caught_speeding.py

@@ -1,21 +1,19 @@
 def caught_speeding(speed, is_birthday):
   speed0 = 60
-  speed1 = 61
-  speed2 = 81
+  speed1 = 80
 
-  if(is_birthday == "yes"):
+  if (is_birthday == True):
     speed0 = speed0 + 5
     speed1 = speed1 + 5
-    speed2 = speed2 + 5
-  
-  if(speed <= speed0):
+    
+  if (speed <= speed0):
     return 0
-  elif(speed1 <= speed < speed2):
+  elif (speed <= speed1):
     return 1
-  elif(speed2 <= speed):
+  else:
     return 2
 
 speed = input("What is your speed? ")
 is_birthday = raw_input("Is it your birthday? (yes or no) ")
 
-print(caught_speeding(speed,is_birthday))
+print(caught_speeding(speed,is_birthday))