Add some tests

Add some tests