small updates on kfold and mlp