As the driver code is in the demo, it should be moved to a suitable location
As the driver code is in the demo, it should be moved to a suitable location