Update ama.py