Completed code and plots

Everything works now, and should match the given plots