Completed the python example