Tiny changes to mouse control