small update to class