tweak Python example