pep8 fix and changed colormap.
pep8 fix and changed colormap.