Add return test cases
More return test cases