UI fixes, adding labels to visualization for clarity