projectile motion with air resistance python