Add a couple of tests
Add a couple of tests