Add some styling to attention histogram.