Added headers to unit tests