Add class for APIController