Refactor disksize module