Update to update the GUI a bit to make everything nicer