Added small code to enable demo