Updated plots for assignment 1