Change init weights randomisation