machine learning script updated