all 4 tests passing
Added all of the tests I can think of