Add models_test.py