Edits to ui