Changing generate weights to update with new argument parameters