Propagated changes to header

Forgot to add them first time around