adding code for plotting figures 1 and 2 from the paper