Update imports to use api module.
Add an auto matplotlib editor.