Added test for model