slight mods to ui