Updated main so it runs the GUI