add missing class definition