Correct init function

Corrected demo initialization function