Changes for edit button and delete class buttons