Updated class name for module export