Added 2 more tests