Improve UI for assert insertion