Adding preprocess function to add header class