improved wgcna script