Modify docstrings a bit
Add docstrings to ui and utils