Improve joystick control