Updated the tests for the kata