Add test cases for Problem 1 and 2.

Also add a few other example functions for inspiration and convenience.