fix example notebook a bit so that it works with current pims