fix up
interface improvment
Merge pull request #69 from Yaliang/master

interface improvement