Python Client Code - Minor modification