From f6c2f152e77e5831434688b1bb609e9d097a685e Mon Sep 17 00:00:00 2001
From: sora10pls <sorafanatic10@gmail.com>
Date: Wed, 6 Dec 2017 18:08:08 -0500
Subject: [PATCH] Misc fixes (pt. 2)

---
 pk3DS.Core/Legality/Legal.cs                | 1 +
 pk3DS.Core/Randomizers/SpeciesRandomizer.cs | 7 +++++++
 2 files changed, 8 insertions(+)

diff --git a/pk3DS.Core/Legality/Legal.cs b/pk3DS.Core/Legality/Legal.cs
index e51489e..b8d93f9 100644
--- a/pk3DS.Core/Legality/Legal.cs
+++ b/pk3DS.Core/Legality/Legal.cs
@@ -225,6 +225,7 @@ public static partial class Legal
             622, 623, 624, 625, 626, 627, 628, 629, 630, 631, 632, 633, 634, 635, 636, 637, 638, 639, 640, 641, 642, 643, 644, 645, 646, 647, 648, 649, 650, 651, 652, 653, 654, 655, 656, 657, 658,
             695, 696, 697, 698, 699, 700, 701, 702, 703,
             719,
+            723, 724, 725, 726, 727, 728
         };
     }
 }
diff --git a/pk3DS.Core/Randomizers/SpeciesRandomizer.cs b/pk3DS.Core/Randomizers/SpeciesRandomizer.cs
index 429dfbc..af7f908 100644
--- a/pk3DS.Core/Randomizers/SpeciesRandomizer.cs
+++ b/pk3DS.Core/Randomizers/SpeciesRandomizer.cs
@@ -193,6 +193,13 @@ private void AddGen7Species(List<int> list)
             list.AddRange(Enumerable.Range(722, 67));
             if (L) list.AddRange(Enumerable.Range(785, 16)); // Tapus, Legends, UBs
             if (E) list.AddRange(Enumerable.Range(801, 2)); // Magearna, Marshadow
+
+            if (MaxSpeciesID == 807) // USUM
+            {
+                if (L) list.AddRange(Enumerable.Range(803, 4)); // USUM UBs
+                if (E) list.AddRange(Enumerable.Range(807, 1)); // Zeraora
+            }
+
         }
 
         public int[] RandomSpeciesList => Enumerable.Range(1, MaxSpeciesID).ToArray();