Adds edit button back