update code
Merge pull request #1 from lndtoan/Branch-1

update code