Python 3 compatibility

I added it.