ML: Integration test completd, to be tested