solved exercise 1, week 4