cleaned up the GUI in the code