updated the code
Merge pull request #2 from Woonyung/master

updated the code