Balloon Flight Prediction Python Code