Added disk usage handling