Part 2 - playing around with layout
Added a button