Small fixes to header file