Added mock classes for better testing.