Trying to reproduce the figure in the paper.