Update headers

Updated comments for consistency