Add astropy.wcs import for wcs_to_celestial_frame