More styling and adding a new button