change mod method to class