Add plots

Add plots and fix a small problem