Removed the naxis stuff.
Now using astropy.io.fits instead of PyFITS