Add expectation for correct answer.
Refactor problem19 solution.