Add test to show usage of custom response class