HW7 Add heading correction