Add tests for using the response_dict