Updated legends to show unit title on edit screens