Improved the edit link functionality.