Small change to tidy up GUI!