Add test to model