Added a parameter to specify data directory.