add Zhang
fix the function.py