diff --git a/scenes/pickaxe_arm.tscn b/scenes/pickaxe_arm.tscn
index 40779e8..ee71e85 100644
--- a/scenes/pickaxe_arm.tscn
+++ b/scenes/pickaxe_arm.tscn
@@ -1,8 +1,10 @@
-[gd_scene load_steps=2 format=3 uid="uid://nusoljma3t55"]
+[gd_scene load_steps=3 format=3 uid="uid://nusoljma3t55"]
 
 [ext_resource type="PackedScene" uid="uid://d1o8g8quh77ll" path="res://assets/pickaxe_arm.glb" id="1_74cmy"]
+[ext_resource type="Script" path="res://scripts/arm.gd" id="2_q356t"]
 
 [node name="PickaxeArm" instance=ExtResource("1_74cmy")]
+script = ExtResource("2_q356t")
 
 [node name="Sphere_004" parent="Skeleton3D" index="0"]
 transform = Transform3D(0.919691, 0, 0, 0, 0.919691, 0, 0, 0, 0.919691, -0.000102931, 10, 6.41946e-09)
diff --git a/scripts/arm.gd b/scripts/arm.gd
index 773ab2f..6a645c8 100644
--- a/scripts/arm.gd
+++ b/scripts/arm.gd
@@ -1,4 +1,4 @@
-extends Node3D
+class_name Arm extends Node3D
 
 @export var action: StringName
 
diff --git a/scripts/body.gd b/scripts/body.gd
index a943207..b043aa1 100644
--- a/scripts/body.gd
+++ b/scripts/body.gd
@@ -1,6 +1,6 @@
-extends Node3D
+class_name Body extends Node3D
 
-@export var arms: Array[Node]
+@export var arms: Array[Arm]
 
 signal arm_enabled(arm)
 signal arm_disabled(arm)