minor changes to class