update with new changes to riemann module