Updated to import by Python 3