Add unit tests for the flight model