updating tests with more scenarios