Implement Problem 2a
More documentation