change import to Python 3