Added logic
Merge pull request #8 from RomanBelkov/master

Added logic