trying to plot stuff for hw 4 Q3