Updated notebook according to paper.