small fixes in the example.py script