Updated for mouse to work in the game