diff --git a/src/Library/CreatureData/CreatureSpecies.hpp b/src/Library/CreatureData/CreatureSpecies.hpp
index ea362c2..09d9e33 100644
--- a/src/Library/CreatureData/CreatureSpecies.hpp
+++ b/src/Library/CreatureData/CreatureSpecies.hpp
@@ -39,6 +39,8 @@ namespace CreatureLib::Library {
         [[nodiscard]] const SpeciesVariant* GetVariant(const std::string& key) const;
         [[nodiscard]] Gender GetRandomGender(Core::Random& rand) const;
         [[nodiscard]] const std::string& GetName() const;
+
+        void SetVariant(const std::string& name, const SpeciesVariant* variant) { _variants[name] = variant; }
     };
 }
 
diff --git a/src/Library/SpeciesLibrary.hpp b/src/Library/SpeciesLibrary.hpp
index ebaf13f..9d95686 100644
--- a/src/Library/SpeciesLibrary.hpp
+++ b/src/Library/SpeciesLibrary.hpp
@@ -26,6 +26,8 @@ namespace CreatureLib::Library {
 
         void LoadSpecies(const std::string& name, const CreatureSpecies* species);
         void DeleteSpecies(const std::string& name);
+
+        size_t GetCount() const { return _species.count(); }
     };
 }