Added 2 test methods