Models actually working

Tested models on terminal. Sorry.