Moved the class to diskutil.py and put it in it's own module and modified the example.py file to reflect this change.