Add joystick code