add test for models