"and the amount as exempt it removes the amount from income also, not just the penalty."

Go to the 8915-E and select if your taxpayer wants to pay all the taxes in 2020 by reporting all of the income in 2020. Otherwise, the standard disaster treatment is the three-year allocation split you see there. Pick one.