cleanup behavior.py