Fix #2 + #5 Make Python 3 compat