Fix class 6 exerise 2