Add missing import to ipython.py