updating the test with more sceanrios