Update initializeWeights.m