Make example.py more user friendly
Add output labels and comment to example.py