final lecture3 fix