Updated docs for training