adding machine learning in python