Started to implement ML for lfi simulated data