Get custom Response class working