Move colorbar generation to colorbar widget