Added setFlight method