Decision tree machine learning assignment