Update causal_parser.py