Added a function that downloads the dataset and extracts the required subset