Feature/model test