Driver code mostly working.