Fix: Directory as input

Now works ok with directories with spaces.