Using Fortran code through f2py