TODO: fix coefficients to make program work with mu != 0