Added to the example notebook