updates to initializeWeights module