Add setHeaders method