Add some tests

