added comment to show where weight initializition takes place