Generalize the plot_cb into a callback function with the additional parameters of the outer and em iterations.
Update args of callback
Add test that prints the full registration objective.