Update from pyfits to astropy.io.fits