added changes in class