Add method to get the drive's total capacity