Fixed a few problems with the GUI