HW7: Update to calculation in *