Add missing headers
API update