Move main() to driver.rs