diff --git a/Demo/Assets/CrashKonijn/GOAP/Demos/Complex/Factories/CleanerAgentTypeConfigFactory.cs b/Demo/Assets/CrashKonijn/GOAP/Demos/Complex/Factories/CleanerAgentTypeConfigFactory.cs
index af1c1e8b..660fc53a 100644
--- a/Demo/Assets/CrashKonijn/GOAP/Demos/Complex/Factories/CleanerAgentTypeConfigFactory.cs
+++ b/Demo/Assets/CrashKonijn/GOAP/Demos/Complex/Factories/CleanerAgentTypeConfigFactory.cs
@@ -9,6 +9,7 @@
 using CrashKonijn.Goap.Demos.Complex.Targets;
 using CrashKonijn.Goap.Demos.Complex.WorldKeys;
 using CrashKonijn.Goap.Runtime;
+using UnityEngine;
 
 namespace CrashKonijn.Goap.Demos.Complex.Factories
 {
@@ -17,7 +18,7 @@ public class CleanerAgentTypeConfigFactory : AgentTypeFactoryBase
         public override IAgentTypeConfig Create()
         {
             var builder = new AgentTypeBuilder(SetIds.Cleaner);
-            
+
             builder.AddCapability<BaseCapability>();
             builder.AddCapability<WanderCapability>();
             builder.AddCapability<HungerCapability>();
@@ -27,13 +28,17 @@ public override IAgentTypeConfig Create()
                 capability.AddGoal<CleanItemsGoal>()
                     .SetBaseCost(20)
                     .AddCondition<ItemsOnFloor>(Comparison.SmallerThanOrEqual, 0);
-                
+
                 capability.AddAction<HaulItemAction>()
                     .SetTarget<HaulTarget>()
                     .AddEffect<ItemsOnFloor>(EffectType.Decrease)
                     .AddCondition<ItemsOnFloor>(Comparison.GreaterThanOrEqual, 1)
-                    .SetMoveMode(ActionMoveMode.PerformWhileMoving);
-                
+                    .SetMoveMode(ActionMoveMode.PerformWhileMoving)
+                    .SetCallback((action) =>
+                    {
+                        Debug.Log($"Action callback: {action}");
+                    });
+
                 capability.AddWorldSensor<ItemOnFloorSensor>()
                     .SetKey<ItemsOnFloor>();
 
diff --git a/Package/Runtime/CrashKonijn.Goap.Core/Interfaces/IActionConfig.cs b/Package/Runtime/CrashKonijn.Goap.Core/Interfaces/IActionConfig.cs
index 12345c0c..a0f64685 100644
--- a/Package/Runtime/CrashKonijn.Goap.Core/Interfaces/IActionConfig.cs
+++ b/Package/Runtime/CrashKonijn.Goap.Core/Interfaces/IActionConfig.cs
@@ -1,4 +1,5 @@
-using CrashKonijn.Agent.Core;
+using System;
+using CrashKonijn.Agent.Core;
 
 namespace CrashKonijn.Goap.Core
 {
@@ -14,4 +15,9 @@ public interface IActionConfig : IClassConfig
         public ActionMoveMode MoveMode { get; }
         public IActionProperties Properties { get; }
     }
-}
\ No newline at end of file
+
+    public interface IClassCallbackConfig
+    {
+        public Action<object> Callback { get; }
+    }
+}
diff --git a/Package/Runtime/CrashKonijn.Goap.Runtime/Classes/AgentTypeFactory.cs b/Package/Runtime/CrashKonijn.Goap.Runtime/Classes/AgentTypeFactory.cs
index 2ba86e04..18e5d845 100644
--- a/Package/Runtime/CrashKonijn.Goap.Runtime/Classes/AgentTypeFactory.cs
+++ b/Package/Runtime/CrashKonijn.Goap.Runtime/Classes/AgentTypeFactory.cs
@@ -64,6 +64,9 @@ private List<IGoapAction> GetActions(IAgentTypeConfig config)
 
             actions.ForEach(x =>
             {
+                if (x.Config is IClassCallbackConfig classCallbackConfig)
+                    classCallbackConfig.Callback?.Invoke(x);
+
                 injector.Inject(x);
                 x.Created();
             });
@@ -79,6 +82,9 @@ private List<IGoal> GetGoals(IAgentTypeConfig config)
 
             goals.ForEach(x =>
             {
+                if (x.Config is IClassCallbackConfig classCallbackConfig)
+                    classCallbackConfig.Callback?.Invoke(x);
+                
                 x.Index = index;
                 index++;
 
@@ -95,6 +101,9 @@ private List<IWorldSensor> GetWorldSensors(IAgentTypeConfig config)
 
             worldSensors.ForEach(x =>
             {
+                if (x.Config is IClassCallbackConfig classCallbackConfig)
+                    classCallbackConfig.Callback?.Invoke(x);
+                
                 injector.Inject(x);
                 x.Created();
             });
@@ -109,6 +118,9 @@ private List<ITargetSensor> GetTargetSensors(IAgentTypeConfig config)
 
             targetSensors.ForEach(x =>
             {
+                if (x.Config is IClassCallbackConfig classCallbackConfig)
+                    classCallbackConfig.Callback?.Invoke(x);
+                
                 injector.Inject(x);
                 x.Created();
             });
@@ -123,6 +135,9 @@ private List<IMultiSensor> GetMultiSensors(IAgentTypeConfig config)
 
             multiSensor.ForEach(x =>
             {
+                if (x.Config is IClassCallbackConfig classCallbackConfig)
+                    classCallbackConfig.Callback?.Invoke(x);
+                
                 injector.Inject(x);
                 x.Created();
             });
diff --git a/Package/Runtime/CrashKonijn.Goap.Runtime/Classes/Builders/ActionBuilder.cs b/Package/Runtime/CrashKonijn.Goap.Runtime/Classes/Builders/ActionBuilder.cs
index cc336b2b..ec0fc286 100644
--- a/Package/Runtime/CrashKonijn.Goap.Runtime/Classes/Builders/ActionBuilder.cs
+++ b/Package/Runtime/CrashKonijn.Goap.Runtime/Classes/Builders/ActionBuilder.cs
@@ -6,80 +6,56 @@
 
 namespace CrashKonijn.Goap.Runtime
 {
-    public class ActionBuilder
+    public class ActionBuilder<T> : ActionBuilder
+        where T : IAction
     {
-        private readonly ActionConfig config;
-        private readonly List<ICondition> conditions = new();
-        private readonly List<IEffect> effects = new();
-        private readonly WorldKeyBuilder worldKeyBuilder;
-        private readonly TargetKeyBuilder targetKeyBuilder;
-        private readonly Type actionType;
-
-        public ActionBuilder(Type actionType, WorldKeyBuilder worldKeyBuilder, TargetKeyBuilder targetKeyBuilder)
-        {
-            this.actionType = actionType;
-            this.worldKeyBuilder = worldKeyBuilder;
-            this.targetKeyBuilder = targetKeyBuilder;
+        public ActionBuilder(WorldKeyBuilder worldKeyBuilder, TargetKeyBuilder targetKeyBuilder) : base(typeof(T), worldKeyBuilder, targetKeyBuilder) { }
 
-            var propType = this.GetPropertiesType();
-
-            this.config = new ActionConfig
-            {
-                Name = actionType.Name,
-                ClassType = actionType.AssemblyQualifiedName,
-                BaseCost = 1,
-                StoppingDistance = 0.5f,
-                RequiresTarget = true,
-                ValidateConditions = true,
-                Properties = (IActionProperties) Activator.CreateInstance(propType),
-            };
-        }
-
-        public ActionBuilder SetTarget<TTargetKey>()
+        public ActionBuilder<T> SetTarget<TTargetKey>()
             where TTargetKey : ITargetKey
         {
             this.config.Target = this.targetKeyBuilder.GetKey<TTargetKey>();
             return this;
         }
 
-        public ActionBuilder SetBaseCost(float baseCost)
+        public ActionBuilder<T> SetBaseCost(float baseCost)
         {
             this.config.BaseCost = baseCost;
             return this;
         }
 
-        public ActionBuilder SetRequiresTarget(bool requiresTarget)
+        public ActionBuilder<T> SetRequiresTarget(bool requiresTarget)
         {
             this.config.RequiresTarget = requiresTarget;
             return this;
         }
 
-        public ActionBuilder SetValidateConditions(bool validate)
+        public ActionBuilder<T> SetValidateConditions(bool validate)
         {
             this.config.ValidateConditions = validate;
             return this;
         }
 
-        public ActionBuilder SetStoppingDistance(float inRange)
+        public ActionBuilder<T> SetStoppingDistance(float inRange)
         {
             this.config.StoppingDistance = inRange;
             return this;
         }
 
         [Obsolete("Use `SetStoppingDistance(float inRange)` instead.")]
-        public ActionBuilder SetInRange(float inRange)
+        public ActionBuilder<T> SetInRange(float inRange)
         {
             this.config.StoppingDistance = inRange;
             return this;
         }
 
-        public ActionBuilder SetMoveMode(ActionMoveMode moveMode)
+        public ActionBuilder<T> SetMoveMode(ActionMoveMode moveMode)
         {
             this.config.MoveMode = moveMode;
             return this;
         }
 
-        public ActionBuilder AddCondition<TWorldKey>(Comparison comparison, int amount)
+        public ActionBuilder<T> AddCondition<TWorldKey>(Comparison comparison, int amount)
             where TWorldKey : IWorldKey
         {
             this.conditions.Add(new Condition
@@ -93,7 +69,7 @@ public ActionBuilder AddCondition<TWorldKey>(Comparison comparison, int amount)
         }
 
         [Obsolete("Use `AddEffect<TWorldKey>(EffectType type)` instead.")]
-        public ActionBuilder AddEffect<TWorldKey>(bool increase)
+        public ActionBuilder<T> AddEffect<TWorldKey>(bool increase)
             where TWorldKey : IWorldKey
         {
             this.effects.Add(new Effect
@@ -105,7 +81,7 @@ public ActionBuilder AddEffect<TWorldKey>(bool increase)
             return this;
         }
 
-        public ActionBuilder AddEffect<TWorldKey>(EffectType type)
+        public ActionBuilder<T> AddEffect<TWorldKey>(EffectType type)
             where TWorldKey : IWorldKey
         {
             this.effects.Add(new Effect
@@ -117,7 +93,7 @@ public ActionBuilder AddEffect<TWorldKey>(EffectType type)
             return this;
         }
 
-        public ActionBuilder SetProperties(IActionProperties properties)
+        public ActionBuilder<T> SetProperties(IActionProperties properties)
         {
             this.ValidateProperties(properties);
 
@@ -125,7 +101,43 @@ public ActionBuilder SetProperties(IActionProperties properties)
             return this;
         }
 
-        private void ValidateProperties(IActionProperties properties)
+        public ActionBuilder<T> SetCallback(Action<T> callback)
+        {
+            this.config.Callback = (obj) => callback((T) obj);
+            return this;
+        }
+    }
+
+    public class ActionBuilder
+    {
+        protected readonly ActionConfig config;
+        protected readonly List<ICondition> conditions = new();
+        protected readonly List<IEffect> effects = new();
+        protected readonly WorldKeyBuilder worldKeyBuilder;
+        protected readonly TargetKeyBuilder targetKeyBuilder;
+        protected readonly Type actionType;
+
+        public ActionBuilder(Type actionType, WorldKeyBuilder worldKeyBuilder, TargetKeyBuilder targetKeyBuilder)
+        {
+            this.actionType = actionType;
+            this.worldKeyBuilder = worldKeyBuilder;
+            this.targetKeyBuilder = targetKeyBuilder;
+
+            var propType = this.GetPropertiesType();
+
+            this.config = new ActionConfig
+            {
+                Name = actionType.Name,
+                ClassType = actionType.AssemblyQualifiedName,
+                BaseCost = 1,
+                StoppingDistance = 0.5f,
+                RequiresTarget = true,
+                ValidateConditions = true,
+                Properties = (IActionProperties) Activator.CreateInstance(propType),
+            };
+        }
+
+        protected void ValidateProperties(IActionProperties properties)
         {
             var actionPropsType = this.GetPropertiesType();
 
@@ -135,7 +147,7 @@ private void ValidateProperties(IActionProperties properties)
             throw new ArgumentException($"The provided properties do not match the expected type '{actionPropsType.Name}'.", nameof(properties));
         }
 
-        private Type GetPropertiesType()
+        protected Type GetPropertiesType()
         {
             var baseType = this.actionType.BaseType;
 
@@ -161,10 +173,10 @@ public IActionConfig Build()
             return this.config;
         }
 
-        public static ActionBuilder Create<TAction>(WorldKeyBuilder worldKeyBuilder, TargetKeyBuilder targetKeyBuilder)
+        public static ActionBuilder<TAction> Create<TAction>(WorldKeyBuilder worldKeyBuilder, TargetKeyBuilder targetKeyBuilder)
             where TAction : IAction
         {
-            return new ActionBuilder(typeof(TAction), worldKeyBuilder, targetKeyBuilder);
+            return new ActionBuilder<TAction>(worldKeyBuilder, targetKeyBuilder);
         }
     }
 }
diff --git a/Package/Runtime/CrashKonijn.Goap.Runtime/Classes/Builders/AgentTypeBuilder.cs b/Package/Runtime/CrashKonijn.Goap.Runtime/Classes/Builders/AgentTypeBuilder.cs
index 7561b210..956a2406 100644
--- a/Package/Runtime/CrashKonijn.Goap.Runtime/Classes/Builders/AgentTypeBuilder.cs
+++ b/Package/Runtime/CrashKonijn.Goap.Runtime/Classes/Builders/AgentTypeBuilder.cs
@@ -97,7 +97,7 @@ public CapabilityBuilder(string name)
             this.capabilityConfig = new CapabilityConfig(name);
         }
 
-        public ActionBuilder AddAction<TAction>()
+        public ActionBuilder<TAction> AddAction<TAction>()
             where TAction : IAction
         {
             var actionBuilder = ActionBuilder.Create<TAction>(this.worldKeyBuilder, this.targetKeyBuilder);
@@ -107,7 +107,7 @@ public ActionBuilder AddAction<TAction>()
             return actionBuilder;
         }
 
-        public GoalBuilder AddGoal<TGoal>()
+        public GoalBuilder<TGoal> AddGoal<TGoal>()
             where TGoal : IGoal
         {
             var goalBuilder = GoalBuilder.Create<TGoal>(this.worldKeyBuilder);
@@ -117,7 +117,7 @@ public GoalBuilder AddGoal<TGoal>()
             return goalBuilder;
         }
 
-        public WorldSensorBuilder AddWorldSensor<TWorldSensor>()
+        public WorldSensorBuilder<TWorldSensor> AddWorldSensor<TWorldSensor>()
             where TWorldSensor : IWorldSensor
         {
             var worldSensorBuilder = WorldSensorBuilder.Create<TWorldSensor>(this.worldKeyBuilder);
@@ -127,7 +127,7 @@ public WorldSensorBuilder AddWorldSensor<TWorldSensor>()
             return worldSensorBuilder;
         }
 
-        public TargetSensorBuilder AddTargetSensor<TTargetSensor>()
+        public TargetSensorBuilder<TTargetSensor> AddTargetSensor<TTargetSensor>()
             where TTargetSensor : ITargetSensor
         {
             var targetSensorBuilder = TargetSensorBuilder.Create<TTargetSensor>(this.targetKeyBuilder);
@@ -137,10 +137,10 @@ public TargetSensorBuilder AddTargetSensor<TTargetSensor>()
             return targetSensorBuilder;
         }
 
-        public MultiSensorBuilder AddMultiSensor<TMultiSensor>()
+        public MultiSensorBuilder<TMultiSensor> AddMultiSensor<TMultiSensor>()
             where TMultiSensor : IMultiSensor
         {
-            var multiSensorBuilder = MultiSensorBuilder.Create<TMultiSensor>(this.worldKeyBuilder);
+            var multiSensorBuilder = MultiSensorBuilder.Create<TMultiSensor>();
 
             this.multiSensorBuilders.Add(multiSensorBuilder);
 
diff --git a/Package/Runtime/CrashKonijn.Goap.Runtime/Classes/Builders/GoalBuilder.cs b/Package/Runtime/CrashKonijn.Goap.Runtime/Classes/Builders/GoalBuilder.cs
index f0c15753..d8c50d76 100644
--- a/Package/Runtime/CrashKonijn.Goap.Runtime/Classes/Builders/GoalBuilder.cs
+++ b/Package/Runtime/CrashKonijn.Goap.Runtime/Classes/Builders/GoalBuilder.cs
@@ -4,45 +4,57 @@
 
 namespace CrashKonijn.Goap.Runtime
 {
-    public class GoalBuilder
+    public class GoalBuilder<T> : GoalBuilder
+        where T : IGoal
     {
-        private readonly GoalConfig config;
-        private readonly List<ICondition> conditions = new();
-        private readonly WorldKeyBuilder worldKeyBuilder;
+        public GoalBuilder(WorldKeyBuilder worldKeyBuilder) : base(typeof(T), worldKeyBuilder) { }
 
-        public GoalBuilder(Type type, WorldKeyBuilder worldKeyBuilder)
-        {
-            this.worldKeyBuilder = worldKeyBuilder;
-            this.config = new GoalConfig(type)
-            {
-                BaseCost = 1,
-                ClassType = type.AssemblyQualifiedName,
-            };
-        }
-
-        public GoalBuilder SetBaseCost(float baseCost)
+        public GoalBuilder<T> SetBaseCost(float baseCost)
         {
             this.config.BaseCost = baseCost;
             return this;
         }
 
-        public GoalBuilder AddCondition<TWorldKey>(Comparison comparison, int amount)
+        public GoalBuilder<T> AddCondition<TWorldKey>(Comparison comparison, int amount)
             where TWorldKey : IWorldKey
         {
             this.conditions.Add(new Condition(this.worldKeyBuilder.GetKey<TWorldKey>(), comparison, amount));
             return this;
         }
 
+        public GoalBuilder<T> SetCallback(Action<T> callback)
+        {
+            this.config.Callback = (obj) => callback((T) obj);
+            return this;
+        }
+    }
+
+    public class GoalBuilder
+    {
+        protected readonly GoalConfig config;
+        protected readonly List<ICondition> conditions = new();
+        protected readonly WorldKeyBuilder worldKeyBuilder;
+
+        public GoalBuilder(Type type, WorldKeyBuilder worldKeyBuilder)
+        {
+            this.worldKeyBuilder = worldKeyBuilder;
+            this.config = new GoalConfig(type)
+            {
+                BaseCost = 1,
+                ClassType = type.AssemblyQualifiedName,
+            };
+        }
+
         public IGoalConfig Build()
         {
             this.config.Conditions = this.conditions;
             return this.config;
         }
 
-        public static GoalBuilder Create<TGoal>(WorldKeyBuilder worldKeyBuilder)
+        public static GoalBuilder<TGoal> Create<TGoal>(WorldKeyBuilder worldKeyBuilder)
             where TGoal : IGoal
         {
-            return new GoalBuilder(typeof(TGoal), worldKeyBuilder);
+            return new GoalBuilder<TGoal>(worldKeyBuilder);
         }
     }
 }
diff --git a/Package/Runtime/CrashKonijn.Goap.Runtime/Classes/Builders/MultiSensorBuilder.cs b/Package/Runtime/CrashKonijn.Goap.Runtime/Classes/Builders/MultiSensorBuilder.cs
index 7f354c85..8536d6ed 100644
--- a/Package/Runtime/CrashKonijn.Goap.Runtime/Classes/Builders/MultiSensorBuilder.cs
+++ b/Package/Runtime/CrashKonijn.Goap.Runtime/Classes/Builders/MultiSensorBuilder.cs
@@ -3,14 +3,24 @@
 
 namespace CrashKonijn.Goap.Runtime
 {
+    public class MultiSensorBuilder<T> : MultiSensorBuilder
+        where T : IMultiSensor
+    {
+        public MultiSensorBuilder() : base(typeof(T)) { }
+        
+        public MultiSensorBuilder<T> SetCallback(Action<T> callback)
+        {
+            this.config.Callback = (obj) => callback((T) obj);
+            return this;
+        }
+    }
+
     public class MultiSensorBuilder
     {
-        private readonly WorldKeyBuilder worldKeyBuilder;
-        private readonly MultiSensorConfig config;
+        protected readonly MultiSensorConfig config;
 
-        public MultiSensorBuilder(Type type, WorldKeyBuilder worldKeyBuilder)
+        public MultiSensorBuilder(Type type)
         {
-            this.worldKeyBuilder = worldKeyBuilder;
             this.config = new MultiSensorConfig()
             {
                 Name = type.Name,
@@ -23,10 +33,10 @@ public IMultiSensorConfig Build()
             return this.config;
         }
 
-        public static MultiSensorBuilder Create<TMultiSensor>(WorldKeyBuilder worldKeyBuilder)
+        public static MultiSensorBuilder<TMultiSensor> Create<TMultiSensor>()
             where TMultiSensor : IMultiSensor
         {
-            return new MultiSensorBuilder(typeof(TMultiSensor), worldKeyBuilder);
+            return new MultiSensorBuilder<TMultiSensor>();
         }
     }
 }
diff --git a/Package/Runtime/CrashKonijn.Goap.Runtime/Classes/Builders/TargetSensorBuilder.cs b/Package/Runtime/CrashKonijn.Goap.Runtime/Classes/Builders/TargetSensorBuilder.cs
index 16ff267d..bddd5c28 100644
--- a/Package/Runtime/CrashKonijn.Goap.Runtime/Classes/Builders/TargetSensorBuilder.cs
+++ b/Package/Runtime/CrashKonijn.Goap.Runtime/Classes/Builders/TargetSensorBuilder.cs
@@ -3,10 +3,30 @@
 
 namespace CrashKonijn.Goap.Runtime
 {
+    public class TargetSensorBuilder<T> : TargetSensorBuilder
+        where T : ITargetSensor
+    {
+        public TargetSensorBuilder(TargetKeyBuilder targetKeyBuilder) : base(typeof(T), targetKeyBuilder) { }
+        
+        public TargetSensorBuilder<T> SetTarget<TTarget>()
+            where TTarget : ITargetKey
+        {
+            this.config.Key = this.targetKeyBuilder.GetKey<TTarget>();
+
+            return this;
+        }
+        
+        public TargetSensorBuilder<T> SetCallback(Action<T> callback)
+        {
+            this.config.Callback = (obj) => callback((T) obj);
+            return this;
+        }
+    }
+    
     public class TargetSensorBuilder
     {
-        private readonly TargetKeyBuilder targetKeyBuilder;
-        private readonly TargetSensorConfig config;
+        protected readonly TargetKeyBuilder targetKeyBuilder;
+        protected readonly TargetSensorConfig config;
 
         public TargetSensorBuilder(Type type, TargetKeyBuilder targetKeyBuilder)
         {
@@ -18,22 +38,14 @@ public TargetSensorBuilder(Type type, TargetKeyBuilder targetKeyBuilder)
             };
         }
 
-        public TargetSensorBuilder SetTarget<TTarget>()
-            where TTarget : ITargetKey
-        {
-            this.config.Key = this.targetKeyBuilder.GetKey<TTarget>();
-
-            return this;
-        }
-
         public ITargetSensorConfig Build()
         {
             return this.config;
         }
 
-        public static TargetSensorBuilder Create<TTargetSensor>(TargetKeyBuilder targetKeyBuilder) where TTargetSensor : ITargetSensor
+        public static TargetSensorBuilder<TTargetSensor> Create<TTargetSensor>(TargetKeyBuilder targetKeyBuilder) where TTargetSensor : ITargetSensor
         {
-            return new TargetSensorBuilder(typeof(TTargetSensor), targetKeyBuilder);
+            return new TargetSensorBuilder<TTargetSensor>(targetKeyBuilder);
         }
     }
 }
diff --git a/Package/Runtime/CrashKonijn.Goap.Runtime/Classes/Builders/WorldSensorBuilder.cs b/Package/Runtime/CrashKonijn.Goap.Runtime/Classes/Builders/WorldSensorBuilder.cs
index 8fdc85e8..f44cf586 100644
--- a/Package/Runtime/CrashKonijn.Goap.Runtime/Classes/Builders/WorldSensorBuilder.cs
+++ b/Package/Runtime/CrashKonijn.Goap.Runtime/Classes/Builders/WorldSensorBuilder.cs
@@ -3,10 +3,30 @@
 
 namespace CrashKonijn.Goap.Runtime
 {
+    public class WorldSensorBuilder<T> : WorldSensorBuilder
+        where T : IWorldSensor
+    {
+        public WorldSensorBuilder(WorldKeyBuilder worldKeyBuilder) : base(typeof(T), worldKeyBuilder) { }
+        
+        public WorldSensorBuilder<T> SetKey<TWorldKey>()
+            where TWorldKey : IWorldKey
+        {
+            this.config.Key = this.worldKeyBuilder.GetKey<TWorldKey>();
+
+            return this;
+        }
+        
+        public WorldSensorBuilder<T> SetCallback(Action<T> callback)
+        {
+            this.config.Callback = (obj) => callback((T) obj);
+            return this;
+        }
+    }
+    
     public class WorldSensorBuilder
     {
-        private readonly WorldKeyBuilder worldKeyBuilder;
-        private readonly WorldSensorConfig config;
+        protected readonly WorldKeyBuilder worldKeyBuilder;
+        protected readonly WorldSensorConfig config;
 
         public WorldSensorBuilder(Type type, WorldKeyBuilder worldKeyBuilder)
         {
@@ -18,23 +38,15 @@ public WorldSensorBuilder(Type type, WorldKeyBuilder worldKeyBuilder)
             };
         }
 
-        public WorldSensorBuilder SetKey<TWorldKey>()
-            where TWorldKey : IWorldKey
-        {
-            this.config.Key = this.worldKeyBuilder.GetKey<TWorldKey>();
-
-            return this;
-        }
-
         public IWorldSensorConfig Build()
         {
             return this.config;
         }
 
-        public static WorldSensorBuilder Create<TWorldSensor>(WorldKeyBuilder worldKeyBuilder)
+        public static WorldSensorBuilder<TWorldSensor> Create<TWorldSensor>(WorldKeyBuilder worldKeyBuilder)
             where TWorldSensor : IWorldSensor
         {
-            return new WorldSensorBuilder(typeof(TWorldSensor), worldKeyBuilder);
+            return new WorldSensorBuilder<TWorldSensor>(worldKeyBuilder);
         }
     }
 }
diff --git a/Package/Runtime/CrashKonijn.Goap.Runtime/Configs/ActionConfig.cs b/Package/Runtime/CrashKonijn.Goap.Runtime/Configs/ActionConfig.cs
index 748965b8..0208bae4 100644
--- a/Package/Runtime/CrashKonijn.Goap.Runtime/Configs/ActionConfig.cs
+++ b/Package/Runtime/CrashKonijn.Goap.Runtime/Configs/ActionConfig.cs
@@ -1,9 +1,10 @@
-using CrashKonijn.Agent.Core;
+using System;
+using CrashKonijn.Agent.Core;
 using CrashKonijn.Goap.Core;
 
 namespace CrashKonijn.Goap.Runtime
 {
-    public class ActionConfig : IActionConfig
+    public class ActionConfig : IActionConfig, IClassCallbackConfig
     {
         public string Name { get; set; }
         public string ClassType { get; set; }
@@ -16,5 +17,6 @@ public class ActionConfig : IActionConfig
         public ICondition[] Conditions { get; set; }
         public IEffect[] Effects { get; set; }
         public ActionMoveMode MoveMode { get; set; } = ActionMoveMode.MoveBeforePerforming;
+        public Action<object> Callback { get; set; }
     }
-}
\ No newline at end of file
+}
diff --git a/Package/Runtime/CrashKonijn.Goap.Runtime/Configs/GoalConfig.cs b/Package/Runtime/CrashKonijn.Goap.Runtime/Configs/GoalConfig.cs
index f8960b3f..fe82b410 100644
--- a/Package/Runtime/CrashKonijn.Goap.Runtime/Configs/GoalConfig.cs
+++ b/Package/Runtime/CrashKonijn.Goap.Runtime/Configs/GoalConfig.cs
@@ -5,7 +5,7 @@
 namespace CrashKonijn.Goap.Runtime
 {
     [Serializable]
-    public class GoalConfig : IGoalConfig
+    public class GoalConfig : IGoalConfig, IClassCallbackConfig
     {
         public GoalConfig() { }
 
@@ -19,6 +19,7 @@ public GoalConfig(Type type)
         public string ClassType { get; set; }
         public float BaseCost { get; set; }
         public List<ICondition> Conditions { get; set; } = new();
+        public Action<object> Callback { get; set; }
 
         public static GoalConfig Create<TGoal>()
             where TGoal : IGoal
diff --git a/Package/Runtime/CrashKonijn.Goap.Runtime/Configs/MultiSensorConfig.cs b/Package/Runtime/CrashKonijn.Goap.Runtime/Configs/MultiSensorConfig.cs
index e2edf5fa..3f240537 100644
--- a/Package/Runtime/CrashKonijn.Goap.Runtime/Configs/MultiSensorConfig.cs
+++ b/Package/Runtime/CrashKonijn.Goap.Runtime/Configs/MultiSensorConfig.cs
@@ -1,10 +1,12 @@
-using CrashKonijn.Goap.Core;
+using System;
+using CrashKonijn.Goap.Core;
 
 namespace CrashKonijn.Goap.Runtime
 {
-    public class MultiSensorConfig : IMultiSensorConfig
+    public class MultiSensorConfig : IMultiSensorConfig, IClassCallbackConfig
     {
         public string Name { get; set; }
         public string ClassType { get; set; }
+        public Action<object> Callback { get; set; }
     }
 }
\ No newline at end of file
diff --git a/Package/Runtime/CrashKonijn.Goap.Runtime/Configs/TargetSensorConfig.cs b/Package/Runtime/CrashKonijn.Goap.Runtime/Configs/TargetSensorConfig.cs
index 0c082bce..875a3ada 100644
--- a/Package/Runtime/CrashKonijn.Goap.Runtime/Configs/TargetSensorConfig.cs
+++ b/Package/Runtime/CrashKonijn.Goap.Runtime/Configs/TargetSensorConfig.cs
@@ -1,4 +1,5 @@
-using CrashKonijn.Goap.Core;
+using System;
+using CrashKonijn.Goap.Core;
 
 namespace CrashKonijn.Goap.Runtime
 {
@@ -18,10 +19,11 @@ public TargetSensorConfig(string name)
         }
     }
 
-    public class TargetSensorConfig : ITargetSensorConfig
+    public class TargetSensorConfig : ITargetSensorConfig, IClassCallbackConfig
     {
         public string Name { get; set; }
         public string ClassType { get; set; }
         public ITargetKey Key { get; set; }
+        public Action<object> Callback { get; set; }
     }
 }
diff --git a/Package/Runtime/CrashKonijn.Goap.Runtime/Configs/WorldSensorConfig.cs b/Package/Runtime/CrashKonijn.Goap.Runtime/Configs/WorldSensorConfig.cs
index 576e8251..ec77aed3 100644
--- a/Package/Runtime/CrashKonijn.Goap.Runtime/Configs/WorldSensorConfig.cs
+++ b/Package/Runtime/CrashKonijn.Goap.Runtime/Configs/WorldSensorConfig.cs
@@ -1,4 +1,5 @@
-using CrashKonijn.Goap.Core;
+using System;
+using CrashKonijn.Goap.Core;
 
 namespace CrashKonijn.Goap.Runtime
 {
@@ -18,10 +19,11 @@ public WorldSensorConfig(string name)
         }
     }
 
-    public class WorldSensorConfig : IWorldSensorConfig
+    public class WorldSensorConfig : IWorldSensorConfig, IClassCallbackConfig
     {
         public string Name { get; set; }
         public string ClassType { get; set; }
         public IWorldKey Key { get; set; }
+        public Action<object> Callback { get; set; }
     }
 }