add solution to week2 assignment