Add more features
Add FlyWithRocket class