add week 2 code.