AddProblemTest refactored