Added solution for week2 assignment1