Updated with initial test case.