Fixed GUI

Cleaned up the GUI