Added headers

Added missing headers to the files.