Python Client Code - minor modification