Update akoa_gen.py