Updated based on clean dataset