Small changes to Fortran code