Basic Machine Learning Model Python Application