Updated the button layout