Added if/else statement for header
Added if/else statement for header