almost done with part 4, just need to plot