Fix Lession 4