add get_disk_size fun for use