diff --git a/src/args.rs b/src/args.rs
index 99b9ae4..a9705a8 100644
--- a/src/args.rs
+++ b/src/args.rs
@@ -162,13 +162,13 @@ pub struct NewUserArgs {
 
 #[derive(Debug, ArgEnum, Copy, Clone, Ord, PartialOrd, Eq, PartialEq, Serialize, Deserialize)]
 pub enum DesktopSetup {
-    #[clap(name = "onxy")]
+    #[clap(name = "onyx")]
     Onyx,
 
     #[clap(name = "gnome")]
     Gnome,
 
-    #[clap(name = "kde")]
+    #[clap(name = "kde", aliases = ["plasma"])]
     Kde,
 
     #[clap(name = "budgie")]
diff --git a/src/functions/locale.rs b/src/functions/locale.rs
index 1527a63..7fa73bc 100755
--- a/src/functions/locale.rs
+++ b/src/functions/locale.rs
@@ -24,6 +24,7 @@ pub fn set_locale(locale: String) {
         files::append_file("/mnt/etc/locale.gen", "en_US.UTF-8 UTF-8\n"),
         "add en_US.UTF-8 UTF-8 to locale.gen",
     );
+    // TODO: Refactor this
     for i in (0..locale.split(' ').count()).step_by(2) {
         files_eval(
             files::append_file(