Adding check for header