Wrote some tests to make sure my code is working. tweaked some things.