Fortran: fixing array size issue in courant()