fix issue #4. both py2 and py3 are supported now.

As per #4