Added some more styling for buttons and box