Added edit functionality

Edit functionality connected to the button