Solved import problems between py2 and py3