Minor changes on header file