test cases now work