add test cases to the code