Minor changes to the GUI