Add training
Update README for training