Feature/model tests

basic tests