Snippet containing implementation of Unit, ProcessUnit and AgentUnit

Unit.cs

public abstract class Unit 
{
    public string Name;
    public State State;

    public abstract void AcceptVisitor(IVisitor visitor);
}

public class ProcessUnit : Unit 
{
    public IEnumerable<Unit> ChildUnits; 

    public override void AcceptVisitor(IVisitor visitor) 
    {
        visitor.VisitProcessUnit(this);
    }
}

public class AgentUnit : Unit 
{
    public void SendCommand(Command command) 
    {
        //Implementation ommited due to simplicity of the example
    }

    public override void AcceptVisitor(IVisitor visitor) 
    {
        visitor.VisitAgentUnit(this);
    }
}