Add some tests
Add some tests