Make it compatible with Python 3