Rewrite header to conform to style, PEP8