Fix contextlib2 import

I've been using your awesome fork, but just noticed this remaining compatibility issue.
Merge pull request #1 from dan-blanchard/patch-1

Fix contextlib2 import