Add a few more tests
Add a few more tests