Update ab_initio.py