simple refactoring
Merge pull request #2 from rudak/master


better later than never ^^