added buttons and some styling