Work on ProbleSet2
working on week3