added simple test cases for program