Test cases now work with hw5 cases