ENH: More tests added.
ENH: Prepare figures for TMI paper.