Reorganize classes and modules.

This reorganization should make it easier to add other colormaps. Related to #8.