Adding week-4-assignment-2