Fixed directory issue

DATA_DIR was incorrectly assigned