Use astropy.wcs to fix headers in extract-image.py

Fixes #11