cleanup code
Use common initialization of the weights