fixed up the gui