improve tran.py