Extending Scatter Plot Node

I would like to extend the Scatter Plot node to include some new functionalities. However, I can't seem to find a NodeModel class for the Scatter Plotter.

Is there a way to extend the Node, or do I have to copy all of the classes into a new project, and do it that way?

Nevermind, I found the classes I needed. They are in org.knime.base.node.viz.plotter.node