Changed header file accordingly