messing around with machine learning alsgorithm