Refactored ShapeCreatorNode so that it takes a function for adding shapes to the model rather than the whole model.