trying to fix ch02 notebook