Updated updateWeight
Updated updateWeight