Python Client Code - Updated