edit function, custom cell class