implimenting machine learning model