fixed implementation
Merge pull request #1 from reekoheek/master

fixed implementation