Updated import path for python3