Add relationship to Response model