Updated class name for API