Added 2 test method