Got the GUI working right