add machine learning class