Fix edit link class after ajax creation. Add TODO