Add the data extraction script