Added code to get files with probabilities and predictions