Add tests

Add some tests.