Add machine learning week 4