Fix examples/01.simple.py import