Got test cases to work