Fixed minor bug in BART.py