Make py3 compatible