Fix before lect 4