ML project assignment