Update modelA.py