First attempt at problem 3.
Add unit testing for problem 01.
Add unit tests for problem 2.