Moved main to Driver