Updating example to extract return value from python.