WIP on adding editing dialog