Small changes to header file