fix python code in README