Add getDiskSize function