Added basic styling
Added some more styling with buttons