Add custom MLP implementation