Added a driver program for the main