Some finetuning for the weights