Adding plots and code for assignment