improve function
Merge pull request #3 from kyaukyuai/master

improve function