Create example.py

refer to this for examples of input format.