Minor changes to GUI