Update UI to use agents, needs a bit of fixing still