Add mouse handling