Final Beta 2.3 adjustments.

2 files changed, 7 insertions, 4 deletions

diff --git a/ShiftOS_TheReturn/Commands.cs b/ShiftOS_TheReturn/Commands.cs
index ce94030..7980635 100644
--- a/ShiftOS_TheReturn/Commands.cs
+++ b/ShiftOS_TheReturn/Commands.cs
@@ -247,7 +247,10 @@ namespace ShiftOS.Engine
         {
             try
             {
-                SaveSystem.CurrentSave.Upgrades[args["upgrade"] as string] = false;
+                SaveSystem.CurrentSave.Upgrades[args["upgrade"].ToString()] = false;
+                SaveSystem.SaveGame();
+                Desktop.PopulateAppLauncher();
+                Desktop.CurrentDesktop.SetupDesktop();
             }
             catch
             {
diff --git a/ShiftOS_TheReturn/Infobox.cs b/ShiftOS_TheReturn/Infobox.cs
index 5900bc4..62abcb7 100644
--- a/ShiftOS_TheReturn/Infobox.cs
+++ b/ShiftOS_TheReturn/Infobox.cs
@@ -65,13 +65,13 @@ namespace ShiftOS.Engine
             });
         }
 
-        public static void PromptText(string title, string message, Action<string> callback)
+        public static void PromptText(string title, string message, Action<string> callback, bool isPassword = false)
         {
             title = Localization.Parse(title);
             message = Localization.Parse(message);
             Desktop.InvokeOnWorkerThread(() =>
             {
-                _infobox.PromptText(title, message, callback);
+                _infobox.PromptText(title, message, callback, isPassword);
             });
         }
 
@@ -99,7 +99,7 @@ namespace ShiftOS.Engine
     public interface IInfobox
     {
         void Open(string title, string msg, Action callback = null);
-        void PromptText(string title, string message, Action<string> callback);
+        void PromptText(string title, string message, Action<string> callback, bool isPassword);
         void PromptYesNo(string title, string message, Action<bool> callback);
     }
 }