Allow output directory through second argument

Fixes #2.