Updated documentation and training script