Add witholding probability argument