playing around with some machine learning