change mlp_layer to layer