Add ability to add/remove step at end, fix coloring with it.