Modification to download Training Set