Add mouse handler
Better styles