Update acording to the code
Merge pull request #13 from iksaku/patch-2

Update acording to the code