Included partb code into driver.cpp