Fixed python code in README