Modify machine learning week 1