Modified plots for visual consistency and colorbar