Modified header file to follow class spec.