change module to class