change from module to class