Adding more UI and doc strings to python code