Added better joystick functionality