Added two lines in two classes