Add test for models