How to add a custom control like RemoveControl?

[qallpat]

Looking to add a custom control like the built in RemoveControl to act as an Edit icon

I've created a class that inherits ExecutableControl like so:

`using Blazor.Diagrams.Core;
using Blazor.Diagrams.Core.Controls;
using Blazor.Diagrams.Core.Events;
using Blazor.Diagrams.Core.Geometry;
using Blazor.Diagrams.Core.Models.Base;
using Blazor.Diagrams.Core.Positions;

namespace DiagramTests.Controls
{
public class EditControl : ExecutableControl
{
private readonly IPositionProvider _positionProvider;

    public EditControl(double x, double y, double offsetX = 0.0, double offsetY = 0.0)
        : this(new BoundsBasedPositionProvider(x, y, offsetX, offsetY))
    {
    }

    public EditControl(IPositionProvider positionProvider)
    {
        _positionProvider = positionProvider;
    }

    public override Point? GetPosition(Model model)
    {
        return _positionProvider.GetPosition(model);
    }

    public override async ValueTask OnPointerDown(Diagram diagram, Model model, PointerEventArgs _)
    {
        // TODO - handle edit click
    }
}

}
`

I had originally assumed this would be a partial, with a razor component to display the edit icon, but looking at the source, I can see RemoveControl has RemoveControlWidget.razor here

Do I need to create the equivalent EditControlWidget.razor?

I have tried adding:

`@using Blazor.Diagrams.Core.Models.Base
@using Blazor.Diagrams.Models

@if (Model is SvgNodeModel)
{
}
else
{

}

@code
{
[Parameter]
public EditControl Control { get; set; } = null!;

[Parameter]
public Model Model { get; set; } = null!;

}`

And I see the error when selecting the node which this control is applied to:

Unhandled exception rendering component: A component couldn't be found for the user action EditControl
Blazor.Diagrams.BlazorDiagramsException: A component couldn't be found for the user action EditControl

[qallpat]

I had forgotten to register the above with the diagram instance!:

Diagram.RegisterComponent<EditControl, EditControlWidget>();