Modified LoadTrainingData to use annotations