Simple Machine Learning Program In Python