Refactor style specific headers code

Allow either a hash or a Style object to set the headers.