Uses directory name if no arguments

Fixes #3