Updated mlp test unit