Add Python 2 support

This was easy... for now.

Fix issue #4