Update assignment

Update 4 plots and the code