Update colors on attention plotter for #64.
Update colors, close #64.