Reorganize classes and modules.

This reorganization should make it easier to add other colormaps.