Small changes to GUI
Small changes to GUI