UI improvements for agents