Adding some tests from homework