Modified getDrives function