Add tests
Add new tests