Fix imports for Python 3