Adding ml homework 5