Cleaning up ballon_flight_test.py