Update docs for pyhton module