Updated the GUI a bit