Added edit button on the widgets