Fixing some import issues.

Also stopped the output from going to output pane and just to REPL
window.