Add stuff to make plotting of MAP estimates work.