Python 3 compatibility (fixes #2)