Added 'download test dataset' functionality