Head unit upgrade