made a little change to GUI