diff --git a/treesize.sh b/treesize.sh
index dd107ac..e0eb502 100755
--- a/treesize.sh
+++ b/treesize.sh
@@ -1,5 +1,6 @@
 #!/usr/bin/env bash
-du -k --max-depth=1 | sort -nr | awk '
+dir=${1:-.}
+du -k --max-depth=1 $dir | sort -nr | awk '
      BEGIN {
         split("KB,MB,GB,TB", Units, ",");
      }