Added custom class for API controller