Added the changes made in class