Fix import error for Python 2.
Bump version to 0.1.2