Added 2 tests