Update smooth_cube.py

Replaced pyfits with astropy