refinemets to weight initialization