Refactor problem 001.
Add problem 002.