updated to class