Should be using game number instead of id to get the pgns. Also it should use the training run id in the path rather than assuming run 1.