Update Driver.cpp

add better functionality to menu