Testing initial code for machine learning set.