Add edit functionality into this project.