real quick code simplification
minor edit
Merge pull request #3 from usefulio/updates-ian

Quick code simplification