Add parallel support using cluster_utils.ipy