Updated to work with later versions of pyes
Merge pull request #3 from faceleg/patch-2

Merged. Thanks a lot!