Add Python syntax to code in README