Update ami.py