Update code.
Merge pull request #1 from pozitr0n/patch-1

Update code.