Played around with layouts and buttons