updated python example