WIP: Python 2 and 3 compatibility

Fix #10