Modified import to be compatible with Python 3