Got the GUI working mostly