Adding mock classes for testing purposes