Updated UI with GridLayout so the buttons are lined up and all the same
size