from Bus import Bus bus1 = Bus(int(input('Enter number of seats in the bus: '))) print(bus1) action = int( input('TAKE Action!\n1 - Passenger in\n2 - Passenger out\n0 - END\n')) print(bus1) while action != 0: if action == 1: bus1.get_on(input('Enter passenger name to get on: ')) elif action == 2: bus1.get_off(input('Enter passenger name to get off: ')) else: print('invalid action.') print(bus1) action = int( input('TAKE Action!\n1 - Passenger in\n2 - Passenger out\n0 - END\n')) print(bus1)