Add better UI for adding robots