optimize aem.py