Add headers to register function
Add headers to register function