added colorbar
try with fewer points to prototype faster