add tokens back in so they're highlighted