Updated the tests features