Remove dependency on RichComparableMixin.
Make imports a little more intuitive.
Checkpointing documentation work.