Add piece class