adding problem from week 5