adding some logic to choosing the drive