working on week 2
problem 1 is done.