edit button now highlights in edit mode

for aesthetic purpose