slight changes to UI