add test for model