Modified code to add a Drive command.