Minor updates

3 files changed, 20 insertions, 21 deletions

diff --git a/src/lib.rs b/src/lib.rs
index 50a3747..b0cf55e 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -1,3 +1,17 @@
+//! # Basic usage
+//!
+//! ```
+//! extern crate diskus;
+//!
+//! use std::path::PathBuf;
+//! use diskus::Walk;
+//!
+//! let num_threads = 4;
+//! let root_directories = &[PathBuf::from(".")];
+//! let walk = Walk::new(root_directories, num_threads);
+//! let size_in_bytes = walk.run();
+//! ```
+
 extern crate crossbeam_channel;
 extern crate humansize;
 extern crate num_cpus;
@@ -5,19 +19,4 @@ extern crate rayon;
 
 pub mod walk;
 
-/// Basic usage of diskus library
-///
-/// # Examples
-///
-/// ```
-/// extern crate diskus;
-/// 
-/// use diskus::Walk;
-/// use std::path::PathBuf;
-/// 
-/// let path = PathBuf::from("/");
-/// let paths = vec![path];
-/// let walk = Walk::new(&paths, 4);
-/// let size = walk.run();
-/// ```
-pub use walk::Walk;
\ No newline at end of file
+pub use walk::Walk;
diff --git a/src/main.rs b/src/main.rs
index 61cf6ee..9dc48d2 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -59,7 +59,7 @@ fn main() {
     let paths: Vec<PathBuf> = matches
         .values_of("path")
         .map(|paths| paths.map(PathBuf::from).collect())
-        .unwrap_or(vec![PathBuf::from(".")]);
+        .unwrap_or_else(|| vec![PathBuf::from(".")]);
 
     let walk = Walk::new(&paths, num_threads);
     let size = walk.run();
diff --git a/src/walk.rs b/src/walk.rs
index e1ccd25..0125713 100644
--- a/src/walk.rs
+++ b/src/walk.rs
@@ -63,14 +63,14 @@ fn walk(tx: channel::Sender<Message>, entries: &[PathBuf]) {
 
 pub struct Walk<'a> {
     root_directories: &'a [PathBuf],
-    threads: usize,
+    num_threads: usize,
 }
 
 impl<'a> Walk<'a> {
-    pub fn new(root_directories: &'a [PathBuf], threads: usize) -> Walk {
+    pub fn new(root_directories: &'a [PathBuf], num_threads: usize) -> Walk {
         Walk {
             root_directories,
-            threads,
+            num_threads,
         }
     }
 
@@ -111,7 +111,7 @@ impl<'a> Walk<'a> {
         });
 
         let pool = rayon::ThreadPoolBuilder::new()
-            .num_threads(self.threads)
+            .num_threads(self.num_threads)
             .build()
             .unwrap();
         pool.install(|| walk(tx, self.root_directories));