Add probem 4.
Make test suite generic.