Add for Problem 3 basic tests that an implementation should pass