Add more functionality to the unit view.