After you file

It can apply to both pension and 401(k) retirement withdrawals. IRA/SEP plans are managed differently. That's how the worksheets in Turbotax are handling it.