Added some styling and buttons