added docs for training