Changes to diskmgmt.py