GUI tidy up code