Adding an optional directory argument to the CLI