Add support for IPython 0.10 and above