Improved UI for inserting element