add .edit-button class