Added Distribution class for the machine learning algorithm