changes to disk_usage to have clearner python code