added model.py