Add some mouse control