Python 3 compatibility

Thanks!