Updated the test cases TDD FTW