Add Edit button and edit handler