Updates to notebook - added visualization.