Added some comments to help understanding of training