diff --git a/src/main.rs b/src/main.rs
index ff8356c..896fe08 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -1,28 +1,25 @@
-use std::fs::OpenOptions;
-use std::path::PathBuf;
-use std::{ffi::OsString, io};
+use std::{ffi::OsString, fs::OpenOptions, io, path::PathBuf};
 
-use musichoard::database::NullDatabase;
-use musichoard::library::{ILibrary, NullLibrary};
-use musichoard::{Collection, MusicHoardBuilder};
 use ratatui::{backend::CrosstermBackend, Terminal};
 use structopt::StructOpt;
 
 use musichoard::{
     database::{
         json::{backend::JsonDatabaseFileBackend, JsonDatabase},
-        IDatabase,
+        IDatabase, NullDatabase,
     },
-    library::beets::{
-        executor::{ssh::BeetsLibrarySshExecutor, BeetsLibraryProcessExecutor},
-        BeetsLibrary,
+    library::{
+        beets::{
+            executor::{ssh::BeetsLibrarySshExecutor, BeetsLibraryProcessExecutor},
+            BeetsLibrary,
+        },
+        ILibrary, NullLibrary,
     },
-    NoLibrary,
+    Collection, MusicHoardBuilder, NoDatabase, NoLibrary,
 };
 
 mod tui;
-use tui::ui::Ui;
-use tui::{event::EventChannel, handler::EventHandler, listener::EventListener, Tui};
+use tui::{event::EventChannel, handler::EventHandler, listener::EventListener, ui::Ui, Tui};
 
 #[derive(StructOpt)]
 struct Opt {
@@ -63,32 +60,9 @@ fn with<LIB: ILibrary, DB: IDatabase>(builder: MusicHoardBuilder<LIB, DB>) {
     Tui::run(terminal, ui, handler, listener).expect("failed to run tui");
 }
 
-fn with_database<DB: IDatabase>(opt: Opt, builder: MusicHoardBuilder<NoLibrary, DB>) {
-    if opt.no_library {
-        with(builder.set_library(NullLibrary));
-    } else if let Some(uri) = opt.beets_ssh_uri {
-        let uri = uri.into_string().expect("invalid SSH URI");
-        let beets_config_file_path = opt
-            .beets_config_file_path
-            .map(|s| s.into_string())
-            .transpose()
-            .expect("failed to extract beets config file path");
-        let lib_exec = BeetsLibrarySshExecutor::new(uri)
-            .expect("failed to initialise beets")
-            .config(beets_config_file_path);
-        with(builder.set_library(BeetsLibrary::new(lib_exec)));
-    } else {
-        let lib_exec = BeetsLibraryProcessExecutor::default().config(opt.beets_config_file_path);
-        with(builder.set_library(BeetsLibrary::new(lib_exec)));
-    }
-}
-
-fn main() {
-    let opt = Opt::from_args();
-    let builder = MusicHoardBuilder::default();
-
+fn with_database<LIB: ILibrary>(opt: Opt, builder: MusicHoardBuilder<LIB, NoDatabase>) {
     if opt.no_database {
-        with_database(opt, builder.set_database(NullDatabase));
+        with(builder.set_database(NullDatabase));
     } else {
         // Create an empty database file if it does not exist.
         match OpenOptions::new()
@@ -109,10 +83,36 @@ fn main() {
         }
 
         let db_exec = JsonDatabaseFileBackend::new(&opt.database_file_path);
-        with_database(opt, builder.set_database(JsonDatabase::new(db_exec)));
+        with(builder.set_database(JsonDatabase::new(db_exec)));
     };
 }
 
+fn with_library(opt: Opt, builder: MusicHoardBuilder<NoLibrary, NoDatabase>) {
+    if opt.no_library {
+        with_database(opt, builder.set_library(NullLibrary));
+    } else if let Some(uri) = opt.beets_ssh_uri {
+        let uri = uri.into_string().expect("invalid SSH URI");
+        let beets_config_file_path = opt
+            .beets_config_file_path
+            .map(|s| s.into_string())
+            .transpose()
+            .expect("failed to extract beets config file path");
+        let lib_exec = BeetsLibrarySshExecutor::new(uri)
+            .expect("failed to initialise beets")
+            .config(beets_config_file_path);
+        with_database(opt, builder.set_library(BeetsLibrary::new(lib_exec)));
+    } else {
+        let lib_exec = BeetsLibraryProcessExecutor::default().config(opt.beets_config_file_path);
+        with_database(opt, builder.set_library(BeetsLibrary::new(lib_exec)));
+    }
+}
+
+fn main() {
+    let opt = Opt::from_args();
+    let builder = MusicHoardBuilder::default();
+    with_library(opt, builder);
+}
+
 #[cfg(test)]
 #[macro_use]
 mod testlib;