Yes go to Form 5329. write the amount of the distribution on Line L and the software will transfer code 12 to Line 2 waiving the penalty. If you look at the instructions of the Form for 2020, code 12 includes Coranavirus Related Distributions and they waive the penalty this year.