Added comment in
Machine learning component