Implemented (correct) edit functionality. Still needs improvement