add new function
Merge pull request #8 from dinghaoDH/master

add new function