Added various test cases

Modified the test file to give different cases