add missing classes for testing