Adding plots and code