Specify the data directory as an argument
Improve task description