Update mymodel.py