Factored out mha parameter