small changes to class name