change code for doing experiment and plotting
how to merge?