changed module to class