Finished problem 1 and wrote tests.