Finished edit and remove functionality.