Cleaned up reflex_base.py