Added edit functionality (not added to the button though)