Check out the updated example of this script here.
Try out this notebook here.
This example also displays the output in full, to show what to expect from a successful run.