Fixed the GUI