Finished up step 5, working on getting pretty plots