Added buttons and some styling