CNN W1 A1 problem with shape

That is not what that comment looks like in my notebook, so you must have modified it. You can get a fresh copy of the notebook using the first topic on the DLS FAQ Thread and compare to what you are seeing.