Add ability to set response type