diff --git a/create_db.go b/create_db.go index c9e6e46..fc395ed 100644 --- a/create_db.go +++ b/create_db.go @@ -25,6 +25,7 @@ var topEnglishWords = []string{ func main() { dbPtr := flag.String("db", "./english_words.db", "SQLite3 DB file to use") + dictFilePtr := flag.String("dict", "./rnd-english", "Dictionary file to use") flag.Parse() // Randomize high Freq. words @@ -67,9 +68,8 @@ func main() { fmt.Println("Top English words inserted into the database.") - // Read the contents of the american-english file - filePath := "./rnd-english" - content, err := ioutil.ReadFile(filePath) + // Read the contents of the shuf american-english file + content, err := ioutil.ReadFile(*dictFilePtr) if err != nil { log.Fatal(err) }