added class in header file