refactored bart to conform to fit & predict API