fixing lection 20