Add mouse handling & update styles