fix lection 3