add doc string
Merge pull request #2 from ZhuoranLyu/master

add doc string