Added colorbar and legend to plots