1 files changed, 17 insertions, 4 deletions
diff --git a/src/Rahm/Desktop/Layout/Draw.hs b/src/Rahm/Desktop/Layout/Draw.hs
index 49921b0..95a34eb 100644
--- a/src/Rahm/Desktop/Layout/Draw.hs
+++ b/src/Rahm/Desktop/Layout/Draw.hs
@@ -6,15 +6,20 @@
 module Rahm.Desktop.Layout.Draw (drawLayout) where
 
 import Control.Arrow (second)
-import Control.Concurrent (threadDelay)
-import Control.Exception (handle)
-import Control.Monad
+import Control.Concurrent ()
+import Control.Exception ()
+import Control.Monad (foldM, forM_, unless)
 import Control.Monad.Writer (execWriter, tell)
 import Data.Foldable (find)
 import Data.Maybe (fromMaybe)
 import Rahm.Desktop.Hash (quickHash)
 import Rahm.Desktop.Layout.Pop (setPop)
 import qualified Rahm.Desktop.StackSet as S
+  ( Screen (..),
+    StackSet (..),
+    Workspace (..),
+    differentiate,
+  )
 import System.Directory (createDirectoryIfMissing, doesFileExist)
 import System.FilePath ((</>))
 import Text.Printf (printf)
@@ -28,7 +33,15 @@ import XMonad
     X,
   )
 import qualified XMonad as X
-import XMonad.Layout.MosaicAlt (expandWindowAlt, shrinkWindowAlt)
+  ( Directories' (cfgDir),
+    LayoutClass (description, handleMessage, runLayout),
+    MonadIO (liftIO),
+    XConf (directories),
+    XState (windowset),
+    asks,
+    gets,
+  )
+import XMonad.Layout.MosaicAlt (expandWindowAlt)
 import XMonad.Layout.Spacing (Border (..), SpacingModifier (..))
 
 -- Draws and returns an XPM for the current layout.