Update python code and README