Added thens to if blocks
Merge pull request #2 from tomatrow/master

Added thens to if blocks