Small GUI changes

Small GUI changes