Adding week-2 assignment-1