Improved joystick functionality