Update qg_basin.py

Now plotting works in parallel