Added import on models, plus fixed typo in line 104/5