A small pythonic change :)
Merge pull request #1 from danbabelnick/patch-1

A small pythonic change