Add tests

Add tests.