Simplicity sake, logic modification
Merge pull request #2 from harrisonde/flow

Simplicity sake, logic modification