made changes in class