Fix README sample python code