Add mouse tracking