Write code for testing 3D flight in small box.