Fixed small bug in function
Merge branch 'master' of https://github.com/oceanhuang/CoZy into feature-functions3