Make compatible with Python 3