Rewrite some class