diff --git a/RustPlugin/Source/RustPlugin/Private/FRustDetailCustomization.cpp b/RustPlugin/Source/RustPlugin/Private/FRustDetailCustomization.cpp
index 8cc30d6..629e478 100644
--- a/RustPlugin/Source/RustPlugin/Private/FRustDetailCustomization.cpp
+++ b/RustPlugin/Source/RustPlugin/Private/FRustDetailCustomization.cpp
@@ -1,7 +1,8 @@
 #include "FRustDetailCustomization.h"
 
 // MyCustomization.cpp
-#include "PropertyEditing.h"
+#include "DetailLayoutBuilder.h"
+#include "DetailWidgetRow.h"
 #include "Widgets/Input/SVectorInputBox.h"
 #include "RustActor.h"
 #include "RustProperty.h"
diff --git a/RustPlugin/Source/RustPlugin/Public/FRustDetailCustomization.h b/RustPlugin/Source/RustPlugin/Public/FRustDetailCustomization.h
index 43a4c62..adddae2 100644
--- a/RustPlugin/Source/RustPlugin/Public/FRustDetailCustomization.h
+++ b/RustPlugin/Source/RustPlugin/Public/FRustDetailCustomization.h
@@ -2,7 +2,6 @@
 #pragma once
 
 #include "IDetailCustomization.h"
-#include "SRustDropdownList.h"
 
 class FRustDetailCustomization: public IDetailCustomization
 {
diff --git a/RustPlugin/Source/RustPlugin/Public/RustProperty.h b/RustPlugin/Source/RustPlugin/Public/RustProperty.h
index c5aba93..8f19999 100644
--- a/RustPlugin/Source/RustPlugin/Public/RustProperty.h
+++ b/RustPlugin/Source/RustPlugin/Public/RustProperty.h
@@ -38,25 +38,25 @@ struct FRustProperty
 	//UPROPERTY(EditAnywhere)
 	//TEnumAsByte<ERustPropertyTag> Tag;
 
-	UPROPERTY(EditAnywhere)
+	UPROPERTY(EditAnywhere, Category=Rust)
 	int32 Tag;
 
-	UPROPERTY(EditAnywhere)
+	UPROPERTY(EditAnywhere, Category=Rust)
 	float Float;
 
-	UPROPERTY(EditAnywhere)
+	UPROPERTY(EditAnywhere, Category=Rust)
 	bool Bool;
 
-	UPROPERTY(EditAnywhere)
+	UPROPERTY(EditAnywhere, Category=Rust)
 	FVector Vector;
 
-	UPROPERTY(EditAnywhere)
+	UPROPERTY(EditAnywhere, Category=Rust)
 	FRotator Rotation;
 
-	UPROPERTY(EditAnywhere)
+	UPROPERTY(EditAnywhere, Category=Rust)
 	TSubclassOf<AActor> Class;
 	
-	UPROPERTY(EditAnywhere)
+	UPROPERTY(EditAnywhere, Category=Rust)
 	TObjectPtr<USoundBase> Sound;
 	static void Initialize(TSharedPtr<IPropertyHandle> Handle, ReflectionType Type);
 };
@@ -67,10 +67,10 @@ struct FDynamicRustComponent
 {
 	GENERATED_BODY()
 
-	UPROPERTY(EditAnywhere)
+	UPROPERTY(EditAnywhere, Category=Rust)
 	TMap<FString, FRustProperty> Fields;
 
-	UPROPERTY(EditAnywhere)
+	UPROPERTY(EditAnywhere, Category=Rust)
 	FString Name;
 
 	void Reload(TSharedPtr<IPropertyHandle> Handle, FGuid Guid);
diff --git a/RustPlugin/Source/RustPlugin/Public/URustReflectionLibrary.h b/RustPlugin/Source/RustPlugin/Public/URustReflectionLibrary.h
index fa825c5..8e89323 100644
--- a/RustPlugin/Source/RustPlugin/Public/URustReflectionLibrary.h
+++ b/RustPlugin/Source/RustPlugin/Public/URustReflectionLibrary.h
@@ -10,15 +10,15 @@ class URustReflectionLibrary: public UBlueprintFunctionLibrary
 	GENERATED_BODY()
 public:
 
-	UFUNCTION(BlueprintCallable)
+	UFUNCTION(BlueprintCallable, Category=Rust)
 	static void K2_GetReflectionVector3(UUuid* Id, FEntity EntityId, int32 Index, FVector &Out);
-	UFUNCTION(BlueprintCallable)
+	UFUNCTION(BlueprintCallable, Category=Rust)
 	static void K2_GetReflectionBool(UUuid* Id, FEntity EntityId, int32 Index, bool &Out);
-	UFUNCTION(BlueprintCallable)
+	UFUNCTION(BlueprintCallable, Category=Rust)
 	static void K2_GetReflectionQuat(UUuid* Id, FEntity EntityId, int32 Index, FQuat &Out);
-	UFUNCTION(BlueprintCallable)
+	UFUNCTION(BlueprintCallable, Category=Rust)
 	static void K2_GetReflectionFloat(UUuid* Id, FEntity EntityId, int32 Index, float &Out);
-	UFUNCTION(BlueprintCallable)
+	UFUNCTION(BlueprintCallable, Category=Rust)
 	static bool K2_HasComponent(UUuid* Id, FEntity EntityId);
 };