Implemented "Edit" button functionality (plot screen)