revise the getting disk space code.