add machine learning src code