Working on week-4-assignment-3