got the plot for prob 1 working