Machine Learning Assignment updates

additional analysis provided