Fix MBSGD stopping criterion.
Clean up visualization.
Choose good setup and add documentation.