Added docstrings and comments to the example.py