editted the GUI a little