Add classes for testing