Added balancing and joystick control