add feature for #3
Merge pull request #4 from icoxfog417/issue3

add feature for #3