Add plot for the paper example