Redesign of the figures in the neighbourhood section of the LMNN notebook

Other minor updates to the notebook as well