Adding a couple more tests