Machine Learning first assignment