Just added the virtual updateUnit method