separating out main from mlp to create different tests