Fixing dust_output.py