Machine Learning Refined Github Solution