change my code after teachter comment