Added headers to fix build errors.