Adding machine learning test algorithm