Added updateUnit methods