updated to specify which halos you want to look at