prototype v2 improvements - improve edit buttons behavior