This is not really just about making your test cases pass. They cared a lot about the code quality. The question is not designed to be overly difficult but it gives you them a good idea about your coding proficiency, so it is not like you can prep for it and go.