Adding initialization to demo