Updating tests
Add the tests