Updated paper.
Updated code for plots.