utils and mlp update