Working on week-5-assignment-4