tidied up the gui