Add tests

This PR adds some tests.