Added 2 new tests