def caught_speeding(speed, is_birthday): extra_speed = 0 if is_birthday: extra_speed = 5 if speed < 61 + extra_speed: return 0 if speed < 81 + extra_speed: return 1 return 2