Add example.py

This adds an executable example so users can easily invoke:

```bash
python example.py
```

and reproduce the output in the screenshot in the README.md