Some minor updates to ui