Added functionality to remove problems.