add button and some styling