Add option to go into ipython on exit