Updated loading for a new training file.