change class name to match with module name