Minor changes to the gui