made changes for class