Update agent.py

add rewrite()