Updated the python documentation in the code and README