Update brain_extraction_dwi.py