Update atmodel.py