Small changes to GUI
Stuck with problem