Fix class 6 exerise 3