Add command line arg to select drive type