Added more plots to assignment