some final changes to the UI layout and Legend