added model test code