minor issues in the example notebook