Add docstring and refactoring for headers
Clean up columns