[FEATURE] Add 'rewrite' argument