Slight changes to the GUI