fix Objective 4