Now should be python3 compatible which closes #2.