diff --git a/main.go b/main.go
index 9b3a861..311faaf 100644
--- a/main.go
+++ b/main.go
@@ -51,10 +51,8 @@ func main() {
 }
 
 func Main() error {
-	var db DB
-	if b, err := os.ReadFile(os.Getenv("DB")); err != nil {
-		return err
-	} else if err := yaml.Unmarshal(b, &db); err != nil {
+	db, err := NewDB()
+	if err != nil {
 		return err
 	}
 	user := os.Getenv("USER")
@@ -89,6 +87,16 @@ func readline() string {
 	return text
 }
 
+func NewDB() (DB, error) {
+	var db DB
+	if b, err := os.ReadFile(os.Getenv("DB")); err != nil {
+		return DB{}, err
+	} else if err := yaml.Unmarshal(b, &db); err != nil {
+		return DB{}, err
+	}
+	return db, nil
+}
+
 func (db DB) HistoryOf(user string) map[string][]History {
 	return db.Users[user].History
 }