Improved the mouse controls