Added weight initialization hooks