added a new test and fixing the old test cases