[MLP/Cost] Add more tests
[MLP/Cost] Simplify API