#4 Add tests.