Working on problem 6.
Wrote most of the plotting code.
Updates to plots to make them look prettier.