updated solution with higher res, plot includes colorbar