diff --git a/home.nix b/home.nix
index 975d57e..85ea8a4 100644
--- a/home.nix
+++ b/home.nix
@@ -25,6 +25,7 @@
       brightnessctl
       curl
       distrobox
+      dig
       fd
       ffmpegthumbnailer
       git
@@ -224,7 +225,6 @@
         tab_powerline_style = "round";
         tab_title_template = "{title}{' :{}:'.format(num_windows) if num_windows > 1 else ''}";
       };
-      # theme = "Catppuccin-Mocha";
     };
     lazygit = {
       enable = true;
@@ -243,15 +243,6 @@
       vimAlias = true;
       viAlias = true;
       clipboard.providers.wl-copy.enable = true;
-      # colorschemes = {
-      #   catppuccin = {
-      #     enable = true;
-      #     settings = {
-      #       flavour = "mocha";
-      #       show_end_of_buffer = true;
-      #     };
-      #   };
-      # };
       globals.mapleader = " ";
       globals.maplocalleader = " ";
       keymaps = [
@@ -291,8 +282,7 @@
         cmp-cmdline.enable = true;
         cmp-fish.enable = true;
         cmp-nvim-lsp.enable = true;
-        cmp-nvim-ultisnips.enable = true;
-        lightline.enable = true;
+        lualine.enable = true;
         fugitive.enable = true;
         gitblame.enable = true;
         gitsigns.enable = true;
@@ -492,8 +482,8 @@
   stylix = {
     image = /home/muhh/Sync/Backgrounds/wald.jpg;
     polarity = "dark";
-    # base16Scheme = "${pkgs.base16-schemes}/share/themes/rose-pine-moon.yaml";
-    base16Scheme = "${pkgs.base16-schemes}/share/themes/catppuccin-mocha.yaml";
+    # base16Scheme = "${pkgs.base16-schemes}/share/themes/catppuccin-mocha.yaml";
+    base16Scheme = "${pkgs.base16-schemes}/share/themes/tokyo-city-terminal-dark.yaml";
   };
 
   wayland.windowManager = {
@@ -560,12 +550,6 @@
         window = {
           titlebar = false;
         };
-        # workspaceOutputAssign = [
-        #   {
-        #     workspace = "2";
-        #     output = "DP-1";
-        #   }
-        # ];
         workspaceOutputAssign = 
           map (x: {
             workspace = x;
diff --git a/hosts/cube.nix b/hosts/cube.nix
index bdc862e..eb8cc50 100644
--- a/hosts/cube.nix
+++ b/hosts/cube.nix
@@ -99,9 +99,11 @@
     shell = pkgs.fish;
     linger = true;
   };
-  # Allow unfree packages
-  nixpkgs.config.allowUnfree = true;
-  nixpkgs.config.allowUnfreePredicate = pkg: true;
+
+  nixpkgs.config = {
+    allowUnfree = true;
+    allowUnfreePredicate = pkg: true;
+  };
 
   environment.shells = with pkgs; [ bash fish zsh ];
   # List packages installed in system profile. To search, run: