Add method to set headers