Added solution for week 2 assignment