update buaa.py, it's working