add model test code