Small updates to training doc