changed version from module to class