Added some sample code for the control flow in python.