Correcting import for Python 3