diff --git a/Program.cs b/Program.cs
index 5738262..f0efe4f 100644
--- a/Program.cs
+++ b/Program.cs
@@ -1,30 +1,118 @@
 using SimsPersonalityGenerator;
 
-var traits = new Dictionary<string, int>
-{
-    { "Outgoing", 0 },
-    { "Nice", 0 },
-    { "Playful", 0 },
-    { "Neat", 0 },
-    { "Active", 0 }
-};
+var isRestartSession = false;
 
-Console.WriteLine("Welcome to the Sim Personality Generator!");
-Console.WriteLine("You can prioritize one or more traits to be above 8 by entering them.");
-Console.Write("Enter traits separated by commas (e.g., Outgoing, Nice), or press Enter to generate a random personality: ");
-var prioritizedTraitsInput = Console.ReadLine()?.Trim();
-
-// Process the prioritized traits
-var prioritizedTraits = new HashSet<string>(prioritizedTraitsInput.Split(',')
-    .Select(t => t.Trim()).Where(t => !string.IsNullOrEmpty(t) && traits.ContainsKey(t)));
-
-if (prioritizedTraits.Count > 0)
+void GenerateTraits()
 {
-    Sim.AssignPrioritizedTraits(traits, prioritizedTraits);
-}
-else
-{
-    Sim.AssignRandomTraits(traits);
+    Console.WriteLine("Welcome to the Sim Personality Generator!");
+    Console.WriteLine("You can prioritize one or more traits to be above 8 by entering them.");
+    Console.Write("Enter traits separated by commas (e.g., Outgoing, Nice), or press Enter to generate a random personality: ");
+    var prioritizedTraitsInput = Console.ReadLine()?.Trim();
+    
+    var traits = new Dictionary<string, int>
+    {
+        { "Outgoing", 0 },
+        { "Nice", 0 },
+        { "Playful", 0 },
+        { "Neat", 0 },
+        { "Active", 0 }
+    };
+
+    // Process the prioritized traits
+    var prioritizedTraits = new HashSet<string>(prioritizedTraitsInput.Split(',')
+        .Select(t => t.Trim()).Where(t => !string.IsNullOrEmpty(t) && traits.ContainsKey(t)));
+
+    if (prioritizedTraits.Count > 0)
+    {
+        Sim.AssignPrioritizedTraits(traits, prioritizedTraits);
+    }
+    else
+    {
+        Sim.AssignRandomTraits(traits);
+    }
+
+    Sim.DisplayPersonality(traits);
+    RestartSession();
 }
 
-Sim.DisplayPersonality(traits);
\ No newline at end of file
+// TODO: Refactor to fix code duplication for both Session methods
+// Method to restart the session based on user input
+void RestartSession()
+{
+    // Prompt the user to decide if they want to try again
+    Console.WriteLine("Would you like to try again?");
+    
+    // Read the user input
+    var input = Console.ReadLine();
+    
+    // Check if the input is null or whitespace
+    if (string.IsNullOrWhiteSpace(input))
+    {
+        // If the input is invalid or empty, exit the application
+        Environment.Exit(Environment.ExitCode);
+    }
+    else
+    {
+        // Convert the input to lowercase to handle case insensitivity
+        switch (input.ToLower())
+        {
+            case "y": // If the user enters 'y'
+                // Set the flag to restart the session
+                isRestartSession = true;
+                // Clear the console for a fresh start
+                Console.Clear();
+                // Call the method to launch a new session
+                LaunchSession();
+                break;
+            default: // If the user enters anything other than 'y'
+                // Exit the application with the current exit code
+                Environment.Exit(Environment.ExitCode);
+                break;
+        }
+    }
+}
+
+// Method to launch a new session
+void LaunchSession()
+{
+    // Check if the session should be restarted
+    if (isRestartSession)
+    {
+        // Reset the restart flag after the session is launched
+        isRestartSession = false;
+        // Generate traits (presumably some form of random data)
+        GenerateTraits();
+    }
+    
+    // Prompt the user to generate a random personality
+    Console.WriteLine("Would you like to generate a random personality? [y/n]");
+    
+    // Read the user input
+    var input = Console.ReadLine();
+    
+    // Check if the input is null or whitespace
+    if (string.IsNullOrWhiteSpace(input))
+    {
+        // If the input is invalid or empty, exit the application
+        Environment.Exit(Environment.ExitCode);
+    }
+    else
+    {
+        // Convert the input to lowercase to handle case insensitivity
+        switch (input.ToLower())
+        {
+            case "y": // If the user enters 'y'
+                // Clear the console for a fresh start
+                Console.Clear();
+                // Generate traits (presumably some form of random data)
+                GenerateTraits();
+                break;
+            default: // If the user enters anything other than 'y'
+                // Exit the application with the current exit code
+                Environment.Exit(Environment.ExitCode);
+                break;
+        }
+    }
+}
+
+LaunchSession();
\ No newline at end of file