slight changes UI change pass