Added the GUI code