small modifications ; increased mlp precision