Added 2 more test methods