Added some important drive enable/disable code for demos