Updated code in class 10/31

Updated code in class 10/31