Add zero indexing option to get_sigma, get_tau in probs.py