Changed pyfits to astropy.io.fits. Minor PEP8 fixes at some places.
Merge pull request #31 from pllim/use-astropy-fits

Changed pyfits to astropy.io.fits.