Minor stylefix, add additional function for customized initial weights