Building a simple machine learning model