WIP: implement batch_engine fit function