Add some imports. Niceness in ipython