Made slight changes to the gui