Use claseses to remove edit button

Not quite as much of a hack