Add 2 more tests