Some small changes to the GUI