Few changes: made it a little more pythonic
minor changed to surprise-me
minor changes to surprise-me
Merge pull request #2 from monty-singh/master

This is more pythonic. Agreed!