added visuals in UI