Fix mhd.py style