Answer to week 1-3 assignment