Is it necessary? It seems you use sigma**2 everywhere below.