Update headers
Changed my mind about how this was set up