duplicate code into week 2