Add some mouse handling code...