Add the piece class