refactor remove the extra function
refactor add wrapper function
Merge pull request #227 from xiangli-cmu/0.2

refactor remove the extra function