Use better weight initialization