Minor changes to main simple flight script